File size: 1,613 Bytes
36d163d 6d86cc8 36d163d 6d86cc8 36d163d |
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 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
job: custom_job
config:
name: flux_train_replicate
process:
- type: custom_sd_trainer
training_folder: output
device: cuda:0
trigger_word: HENDRIK
network:
type: lora
linear: 128
linear_alpha: 128
save:
dtype: float16
save_every: 100
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: 2
steps: 1600
gradient_accumulation_steps: 1
train_unet: true
train_text_encoder: false
content_or_style: balanced
gradient_checkpointing: true
noise_scheduler: flowmatch
optimizer: adamw
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: true
sample:
sampler: flowmatch
sample_every: 100
width: 1024
height: 1024
prompts:
- HENDRIK digital photograph with indoor lighting of man standing. The man is
wearing a black shiny sleeveless vinyl trench coat that extends down to the
knees, bright pink gloves on their hands. The person also has on black sturdy boots.
His face is partially obscured by glasses and a short haircut.
neg: ''
seed: 42
walk_seed: true
guidance_scale: 3.5
sample_steps: 28
meta:
name: flux_train_replicate
version: '1.0'
|