Instructions to use Mavlon001/stylee-diffusion with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use Mavlon001/stylee-diffusion with PEFT:
Task type is invalid.
- Notebooks
- Google Colab
- Kaggle
File size: 966 Bytes
3a4fc8b cf44389 3a4fc8b cf44389 3a4fc8b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | {
"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"
} |