test-task4 / checkpoints /config.yaml
Jordansky's picture
Upload task output 956b14d3-7c4d-4d1a-8614-19d6bd025b2f
b64b852 verified
Raw
History Blame Contribute Delete
1.32 kB
config:
name: last
process:
- datasets:
- cache_latents_to_disk: true
caption_dropout_rate: 0.3
caption_ext: txt
folder_path: /dataset/images/956b14d3-7c4d-4d1a-8614-19d6bd025b2f/img
is_reg: false
resolution:
- 512
- 768
- 1024
shuffle_tokens: false
device: cuda
model:
arch: zimage:turbo
assistant_lora_path: /opt/adapters/zimage/zimage_turbo_training_adapter_v2.safetensors
name_or_path: /cache/models/gradients-io-tournaments--Z-Image-Turbo
qtype: qfloat8
qtype_te: qfloat8
quantize: true
quantize_te: true
network:
linear: 16
linear_alpha: 16
type: lora
save:
dtype: bf16
max_step_saves_to_keep: 4
save_every: 200
save_format: diffusers
train:
batch_size: 1
cache_text_embeddings: false
dtype: bf16
ema_config:
ema_decay: 0.995
use_ema: false
use_num_updates: true
gradient_checkpointing: true
lr: 0.0001
noise_scheduler: flowmatch
optimizer: adamw8bit
steps: 1610
timestep_type: sigmoid
training_folder: /app/checkpoints/956b14d3-7c4d-4d1a-8614-19d6bd025b2f/test-task4
type: diffusion_trainer
job: extension
meta:
name: zimage_lora
version: '1.0'