test_zimage-1 / checkpoints /last /config.yaml
Jordansky's picture
Upload task output f9f60b7d-7df2-4bfc-a394-e565449a2138
60048c5 verified
config:
name: last
process:
- datasets:
- cache_latents_to_disk: true
caption_ext: txt
folder_path: /dataset/images/f9f60b7d-7df2-4bfc-a394-e565449a2138/img
is_reg: false
resolution:
- 512
- 768
- 1024
device: cuda
model:
arch: zimage:turbo
assistant_lora_path: /cache/hf_cache/zimage_turbo_training_adapter_v2.safetensors
name_or_path: /cache/models/gradients-io-tournaments--Z-Image-Turbo
qtype: qfloat8
qtype_te: qfloat8
quantize: false
quantize_te: false
network:
conv: 32
conv_alpha: 32
linear: 128
linear_alpha: 64
type: lora
save:
dtype: bf16
max_step_saves_to_keep: 4
save_every: 250
save_format: diffusers
train:
batch_size: 1
dtype: bf16
gradient_accumulation: 4
gradient_checkpointing: true
lr: 0.00018
noise_scheduler: flowmatch
optimizer: adamw8bit
steps: 1100
timestep_type: weighted
training_folder: /app/checkpoints/f9f60b7d-7df2-4bfc-a394-e565449a2138/test_zimage-1
type: diffusion_trainer
job: extension
meta:
name: zimage_lora
version: '1.0'