test-task1 / checkpoints /config.yaml
Jordansky's picture
Upload task output f8e8c67b-4686-49f2-803d-313f4437b93d
641986d verified
Raw
History Blame Contribute Delete
1.38 kB
config:
name: last
process:
- datasets:
- cache_latents_to_disk: true
caption_dropout_rate: 0.0
caption_ext: txt
folder_path: /dataset/images/f8e8c67b-4686-49f2-803d-313f4437b93d/img
is_reg: false
resolution:
- 512
- 768
- 1024
shuffle_tokens: false
device: cuda
model:
arch: qwen_image
low_vram: false
name_or_path: /cache/models/gradients-io-tournaments--Qwen-Image-Jib-Mix
qtype: float8
quantize: true
quantize_te: true
network:
linear: 32
linear_alpha: 32
type: lora
save:
dtype: bf16
max_step_saves_to_keep: 4
save_every: 100
save_format: diffusers
train:
batch_size: 1
cache_text_embeddings: true
cfg_scale: 6.0
do_cfg: true
dtype: bf16
ema_config:
ema_decay: 0.995
use_ema: false
gradient_accumulation: 1
gradient_checkpointing: true
lr: 0.00011
noise_scheduler: flowmatch
optimizer: adamw8bit
optimizer_params:
weight_decay: 6.0e-06
steps: 1147
timestep_type: lognorm_blend
train_text_encoder: false
train_unet: true
training_folder: /app/checkpoints/f8e8c67b-4686-49f2-803d-313f4437b93d/test-task1
type: diffusion_trainer
job: extension
meta:
name: qwen_image_lora
version: '1.0'