| { | |
| "pretrained_model_name_or_path": "runwayml/stable-diffusion-inpainting", | |
| "pretrained_vae_name_or_path": "stabilityai/sd-vae-ft-mse", | |
| "revision": "fp16", | |
| "tokenizer_name": null, | |
| "instance_data_dir": null, | |
| "class_data_dir": null, | |
| "instance_prompt": null, | |
| "class_prompt": null, | |
| "save_sample_prompt": "woman wearing a blue sks bikini top with qsd pattern|woman wearing a blue sks bikini bottom with qsd pattern", | |
| "save_sample_negative_prompt": null, | |
| "n_save_sample": 4, | |
| "save_guidance_scale": 7.5, | |
| "save_infer_steps": 50, | |
| "pad_tokens": true, | |
| "with_prior_preservation": true, | |
| "prior_loss_weight": 1.0, | |
| "num_class_images": 100, | |
| "output_dir": "/home/fant/repos/shiv-diffusers/output/model", | |
| "seed": 3434554, | |
| "resolution": 512, | |
| "center_crop": false, | |
| "train_text_encoder": true, | |
| "train_batch_size": 1, | |
| "sample_batch_size": 4, | |
| "num_train_epochs": 20, | |
| "max_train_steps": 4000, | |
| "gradient_accumulation_steps": 1, | |
| "gradient_checkpointing": false, | |
| "learning_rate": 1e-06, | |
| "scale_lr": false, | |
| "lr_scheduler": "constant", | |
| "lr_warmup_steps": 0, | |
| "use_8bit_adam": false, | |
| "adam_beta1": 0.9, | |
| "adam_beta2": 0.999, | |
| "adam_weight_decay": 0.01, | |
| "adam_epsilon": 1e-08, | |
| "max_grad_norm": 1.0, | |
| "push_to_hub": false, | |
| "hub_token": null, | |
| "hub_model_id": null, | |
| "logging_dir": "logs", | |
| "log_interval": 10, | |
| "save_interval": 800, | |
| "save_min_steps": 0, | |
| "mixed_precision": "fp16", | |
| "not_cache_latents": true, | |
| "local_rank": -1, | |
| "concepts_list": [ | |
| { | |
| "url": "https://www.brightswimwear.com/products/bali-top-ocean-dream", | |
| "instance_prompt": "woman wearing a blue sks bikini top with qsd pattern", | |
| "image_urls": [ | |
| "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/top/0.png", | |
| "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/top/1.png", | |
| "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/top/2.png", | |
| "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/top/3.png" | |
| ], | |
| "class_prompt": "woman wearing a bikini top", | |
| "class_data_dir": "./data/class/woman-wearing-a-bikini-top", | |
| "instance_data_dir": "./data/instance/woman-wearing-a-blue-sks-bikini-top-with-qsd-pattern" | |
| }, | |
| { | |
| "url": "https://www.brightswimwear.com/products/bali-bottom-ocean-dream", | |
| "instance_prompt": "woman wearing a blue sks bikini bottom with qsd pattern", | |
| "image_urls": [ | |
| "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/bottom/0.png", | |
| "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/bottom/1.png", | |
| "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/bottom/2.png", | |
| "https://storage.googleapis.com/dreambooth-training/bikini-crops/bali-ocean-dream/bottom/3.png" | |
| ], | |
| "class_prompt": "woman wearing a bikini bottom", | |
| "class_data_dir": "./data/class/woman-wearing-a-bikini-bottom", | |
| "instance_data_dir": "./data/instance/woman-wearing-a-blue-sks-bikini-bottom-with-qsd-pattern" | |
| } | |
| ] | |
| } |