job: custom_job config: name: flux_train_replicate process: - type: custom_sd_trainer training_folder: output device: cuda:0 trigger_word: USER_295133871 network: type: lora linear: 128 linear_alpha: 128 network_kwargs: only_if_contains: - transformer.single_transformer_blocks.0.proj_out - transformer.single_transformer_blocks.1.proj_out - transformer.single_transformer_blocks.2.proj_out - transformer.single_transformer_blocks.3.proj_out - transformer.single_transformer_blocks.4.proj_out - transformer.single_transformer_blocks.5.proj_out - transformer.single_transformer_blocks.6.proj_out - transformer.single_transformer_blocks.7.proj_out - transformer.single_transformer_blocks.8.proj_out - transformer.single_transformer_blocks.9.proj_out - transformer.single_transformer_blocks.10.proj_out - transformer.single_transformer_blocks.11.proj_out - transformer.single_transformer_blocks.12.proj_out - transformer.single_transformer_blocks.13.proj_out - transformer.single_transformer_blocks.14.proj_out - transformer.single_transformer_blocks.15.proj_out - transformer.single_transformer_blocks.16.proj_out - transformer.single_transformer_blocks.17.proj_out - transformer.single_transformer_blocks.18.proj_out - transformer.single_transformer_blocks.19.proj_out - transformer.single_transformer_blocks.20.proj_out - transformer.single_transformer_blocks.21.proj_out - transformer.single_transformer_blocks.22.proj_out - transformer.single_transformer_blocks.23.proj_out - transformer.single_transformer_blocks.24.proj_out - transformer.single_transformer_blocks.25.proj_out - transformer.single_transformer_blocks.26.proj_out - transformer.single_transformer_blocks.27.proj_out - transformer.single_transformer_blocks.28.proj_out - transformer.single_transformer_blocks.29.proj_out - transformer.single_transformer_blocks.30.proj_out - transformer.single_transformer_blocks.31.proj_out - transformer.single_transformer_blocks.32.proj_out - transformer.single_transformer_blocks.33.proj_out - transformer.single_transformer_blocks.34.proj_out - transformer.single_transformer_blocks.35.proj_out - transformer.single_transformer_blocks.36.proj_out - transformer.single_transformer_blocks.37.proj_out save: dtype: float16 save_every: 1201 max_step_saves_to_keep: 1 datasets: - folder_path: input_images caption_ext: txt caption_dropout_rate: 0.05 shuffle_tokens: false cache_latents_to_disk: false cache_latents: true resolution: - 512 - 768 - 1024 train: batch_size: 1 steps: 1200 gradient_accumulation_steps: 1 train_unet: true train_text_encoder: false content_or_style: balanced gradient_checkpointing: false noise_scheduler: flowmatch optimizer: adamw8bit lr: 0.0004 ema_config: use_ema: true ema_decay: 0.99 dtype: bf16 model: name_or_path: FLUX.1-dev is_flux: true quantize: false sample: sampler: flowmatch sample_every: 1201 width: 1024 height: 1024 prompts: [] neg: '' seed: 42 walk_seed: true guidance_scale: 3.5 sample_steps: 28 meta: name: flux_train_replicate version: '1.0'