| { | |
| "base_model": "runwayml/stable-diffusion-v1-5", | |
| "output_dir": "/home/zeus/stylee_diffusion", | |
| "lora_r": 16, | |
| "lora_alpha": 32, | |
| "lora_dropout": 0.05, | |
| "lora_target_modules": [ | |
| "to_q", | |
| "to_v", | |
| "to_k", | |
| "to_out.0" | |
| ], | |
| "epochs": 5, | |
| "batch_size": 1, | |
| "gradient_accumulation": 4, | |
| "learning_rate": 0.0001, | |
| "lr_scheduler": "cosine", | |
| "warmup_steps": 100, | |
| "max_grad_norm": 1.0, | |
| "mixed_precision": "fp16", | |
| "save_steps": 100, | |
| "logging_steps": 20, | |
| "seed": 42, | |
| "image_size": 512, | |
| "center_crop": true, | |
| "num_inference_steps": 50, | |
| "guidance_scale": 7.5, | |
| "negative_prompt": "blurry, low quality, bad anatomy, deformed, ugly, unrealistic, western fashion, non-African context, extra limbs, bad proportions", | |
| "data_dir": "/home/zeus/african_fashion_images", | |
| "hf_repo_id": "your-username/stylee-diffusion", | |
| "push_to_hub": true, | |
| "save_to_drive": false, | |
| "drive_path": "/content/drive/MyDrive/stylee_ai/stylee_diffusion" | |
| } |