| { |
| "pretrained_model_name_or_path": "stabilityai/stable-diffusion-2-1-base", |
| "revision": "fp16", |
| "instance_data_dir": "images/mushrooms", |
| "class_data_dir": "images/doy pack", |
| "instance_prompt": "xyz doy pack", |
| "class_prompt": "a doy pack", |
| "save_sample_prompt": null, |
| "save_sample_negative_prompt": null, |
| "n_save_sample": 5, |
| "save_guidance_scale": 7.5, |
| "save_infer_steps": 50, |
| "pad_tokens": false, |
| "prior_loss_weight": 1.0, |
| "num_class_images": 300, |
| "output_dir": "results", |
| "seed": 1234, |
| "resolution": 512, |
| "center_crop": false, |
| "train_batch_size": 1, |
| "sample_batch_size": 5, |
| "max_train_steps": 3000, |
| "max_train_text_encoder": 1500, |
| "gradient_checkpointing": false, |
| "learning_rate": 1e-06, |
| "lr_scheduler": "constant", |
| "lr_warmup_steps": 500, |
| "adam_beta1": 0.9, |
| "adam_beta2": 0.999, |
| "adam_weight_decay": 0.01, |
| "adam_epsilon": 1e-08, |
| "max_grad_norm": 1.0, |
| "log_interval": 10, |
| "save_interval": 10000, |
| "save_min_steps": 5000, |
| "mixed_precision": "fp16", |
| "concepts_list": [ |
| { |
| "instance_prompt": "xyz doy pack", |
| "class_prompt": "a doy pack", |
| "instance_data_dir": "images/mushrooms", |
| "class_data_dir": "images/doy pack" |
| } |
| ], |
| "repo_name": "mush", |
| "instance_class": "doy pack", |
| "num_train_epochs": 10 |
| } |
