finaltest-FLUX / checkpoints /config.yaml
Gege24's picture
Upload task output 8b838fd0-24cf-4f2c-86bb-d9f4fb44b792
1804372 verified
Raw
History Blame Contribute Delete
1.3 kB
config:
name: last
process:
- datasets:
- cache_latents_to_disk: true
caption_dropout_rate: 0.04
caption_ext: txt
folder_path: /dataset/images/8b838fd0-24cf-4f2c-86bb-d9f4fb44b792/img
is_reg: false
resolution:
- 512
- 768
- 1024
device: cuda
model:
is_flux: true
name_or_path: /cache/models/black-forest-labs--FLUX.1-dev
qtype: qfloat8
qtype_te: qfloat8
quantize: false
quantize_te: false
network:
linear: 48
linear_alpha: 48
type: lora
save:
dtype: bf16
max_step_saves_to_keep: 12
save_every: 250
save_format: diffusers
train:
batch_size: 1
content_or_style: content
disable_sampling: true
dtype: bf16
ema_config:
ema_decay: 0.999
use_ema: false
gradient_checkpointing: true
lr: 0.0001
noise_scheduler: flowmatch
optimizer: adamw8bit
skip_first_sample: true
steps: 2400
timestep_type: weighted
train_text_encoder: false
train_unet: true
training_folder: /app/checkpoints/8b838fd0-24cf-4f2c-86bb-d9f4fb44b792/finaltest-FLUX
trigger_word: a 3dicon
type: diffusion_trainer
job: extension
meta:
name: flux_lora
version: '1.0'