| { | |
| "base_model": "black-forest-labs/FLUX.1-dev", | |
| "dataset": "dolphinium/wikiart-impressionism-curated", | |
| "training_config": { | |
| "job": "extension", | |
| "config": { | |
| "name": "flux_1_dev_wikiart_impressionism", | |
| "process": [ | |
| { | |
| "type": "sd_trainer", | |
| "training_folder": "/content/output", | |
| "device": "cuda:0", | |
| "network": { | |
| "type": "lora", | |
| "linear": 16, | |
| "linear_alpha": 16 | |
| }, | |
| "save": { | |
| "dtype": "float16", | |
| "save_every": 250, | |
| "max_step_saves_to_keep": 4 | |
| }, | |
| "datasets": [ | |
| { | |
| "folder_path": "/content/dataset", | |
| "caption_ext": "txt", | |
| "caption_dropout_rate": 0.05, | |
| "shuffle_tokens": true, | |
| "cache_latents_to_disk": true, | |
| "resolution": [ | |
| 512, | |
| 768, | |
| 1024 | |
| ] | |
| } | |
| ], | |
| "train": { | |
| "batch_size": 1, | |
| "steps": 2000, | |
| "gradient_accumulation_steps": 1, | |
| "train_unet": true, | |
| "train_text_encoder": false, | |
| "content_or_style": "balanced", | |
| "gradient_checkpointing": true, | |
| "noise_scheduler": "flowmatch", | |
| "optimizer": "adamw8bit", | |
| "lr": 0.0001, | |
| "ema_config": { | |
| "use_ema": true, | |
| "ema_decay": 0.99 | |
| }, | |
| "dtype": "bf16" | |
| }, | |
| "model": { | |
| "name_or_path": "black-forest-labs/FLUX.1-dev", | |
| "is_flux": true, | |
| "quantize": true | |
| }, | |
| "sample": { | |
| "sampler": "flowmatch", | |
| "sample_every": 250, | |
| "width": 1024, | |
| "height": 1024, | |
| "prompts": [ | |
| "an impressionist style landscape with rolling hills and autumn trees", | |
| "a portrait of a woman in impressionist style, soft lighting", | |
| "an urban scene with cafe terraces in impressionist style", | |
| "a still life with flowers in impressionist painting style" | |
| ], | |
| "neg": "", | |
| "seed": 42, | |
| "walk_seed": true, | |
| "guidance_scale": 4, | |
| "sample_steps": 20 | |
| } | |
| } | |
| ] | |
| }, | |
| "meta": { | |
| "name": "[name]", | |
| "version": "1.0" | |
| } | |
| } | |
| } |