| { | |
| "data_dir": "training_data", | |
| "output_dir": "lora_output_v2", | |
| "num_epochs": 100, | |
| "learning_rate": 1e-05, | |
| "weight_decay": 0.01, | |
| "gradient_clip": 1.0, | |
| "seed": 42, | |
| "warmup_epochs": 5, | |
| "gradient_accumulation_steps": 16, | |
| "lora_rank": 32, | |
| "lora_alpha": 64, | |
| "lora_dropout": 0.05, | |
| "save_every": 20, | |
| "validate_every": 5, | |
| "model_id": "stabilityai/stable-diffusion-xl-base-1.0", | |
| "prompt": "a high quality photograph", | |
| "num_timestep_buckets": 8, | |
| "max_timestep": 600 | |
| } |