| { |
| "model_type": "stable-diffusion", |
| "config": { |
| "name": "flux_train_replicate", |
| "process": [ |
| { |
| "datasets": [ |
| { |
| "cache_latents_to_disk": true, |
| "caption_dropout_rate": 0.05, |
| "caption_ext": "txt", |
| "folder_path": "input_images", |
| "resolution": [ |
| 512, |
| 768, |
| 1024 |
| ], |
| "shuffle_tokens": false |
| } |
| ], |
| "device": "cuda:0", |
| "model": { |
| "is_flux": true, |
| "name_or_path": "black-forest-labs/FLUX.1-dev", |
| "quantize": true |
| }, |
| "network": { |
| "linear": 16, |
| "linear_alpha": 16, |
| "type": "lora" |
| }, |
| "sample": { |
| "guidance_scale": 4, |
| "height": 1024, |
| "neg": "", |
| "prompts": [ |
| "a sign that says 'I LOVE PROMPTS!' in the style of [trigger]" |
| ], |
| "sample_every": 250, |
| "sample_steps": 20, |
| "sampler": "flowmatch", |
| "seed": 42, |
| "walk_seed": true, |
| "width": 1024 |
| }, |
| "save": { |
| "dtype": "float16", |
| "max_step_saves_to_keep": 1, |
| "save_every": 1001 |
| }, |
| "train": { |
| "batch_size": 1, |
| "content_or_style": "balanced", |
| "dtype": "bf16", |
| "ema_config": { |
| "ema_decay": 0.99, |
| "use_ema": true |
| }, |
| "gradient_accumulation_steps": 1, |
| "gradient_checkpointing": true, |
| "lr": 0.0004, |
| "noise_scheduler": "flowmatch", |
| "optimizer": "adamw8bit", |
| "steps": 1000, |
| "train_text_encoder": false, |
| "train_unet": true |
| }, |
| "training_folder": "output", |
| "trigger_word": "TOK", |
| "type": "sd_trainer" |
| } |
| ] |
| }, |
| "job": "extension", |
| "meta": { |
| "name": "flux_train_replicate", |
| "version": "1.0" |
| } |
| } |
|
|
