stylee_config.json

{
  "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"
}