job: extension config: name: kontext_remove process: - type: sd_trainer training_folder: output device: cuda:0 network: type: lora linear: 16 linear_alpha: 16 save: dtype: float16 save_every: 250 max_step_saves_to_keep: 4 push_to_hub: false datasets: - folder_path: datasets/kontext_training/images control_path: datasets/kontext_training/controls caption_ext: txt caption_dropout_rate: 0.05 shuffle_tokens: false cache_latents_to_disk: true resolution: - 512 - 768 - 1024 train: batch_size: 1 steps: 4000 gradient_accumulation_steps: 1 train_unet: true train_text_encoder: false gradient_checkpointing: true noise_scheduler: flowmatch optimizer: adamw8bit lr: 0.0001 timestep_type: weighted dtype: bf16 model: name_or_path: models/FLUX.1-Kontext-dev arch: flux_kontext quantize: true sample: sampler: flowmatch sample_every: 250 width: 1024 height: 1024 prompts: - Remove red SUV with luggage on roof rack. Vast, flat, white salt flat landscape under a clear blue sky with distant mountains on the horizon. --ctrl_img datasets/kontext_training/controls/object_removal_01.png - Remove blurry background. A laptop keyboard is visible at the top of the frame, with a Samsung 1TB NVMe SSD lying on a light brown surface below it. --ctrl_img datasets/kontext_training/controls/object_removal_02.png - Remove snowflake. A window frame is visible on the left and right sides of a dark interior space, with bright light coming from outside the window. --ctrl_img datasets/kontext_training/controls/object_removal_03.png neg: '' seed: 42 walk_seed: true guidance_scale: 4 sample_steps: 20 meta: name: kontext_remove version: '1.0'