bimabk commited on
Commit
d7eb4a3
·
verified ·
1 Parent(s): 1fb976c

Upload task output 1c93dd95-2e89-48d9-813d-e0f521599cfd

Browse files
Files changed (1) hide show
  1. checkpoints/last/config.yaml +50 -0
checkpoints/last/config.yaml ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ config:
2
+ name: last
3
+ process:
4
+ - datasets:
5
+ - cache_latents_to_disk: true
6
+ caption_ext: txt
7
+ folder_path: /dataset/images/1c93dd95-2e89-48d9-813d-e0f521599cfd/img
8
+ is_reg: false
9
+ resolution:
10
+ - 512
11
+ - 768
12
+ - 1024
13
+ device: cuda
14
+ model:
15
+ arch: zimage:turbo
16
+ assistant_lora_path: /cache/hf_cache/zimage_turbo_training_adapter_v2.safetensors
17
+ name_or_path: /cache/models/gradients-io-tournaments--Z-Image-Turbo
18
+ qtype: qfloat8
19
+ qtype_te: qfloat8
20
+ quantize: true
21
+ quantize_te: true
22
+ network:
23
+ conv: 16
24
+ conv_alpha: 16
25
+ linear: 32
26
+ linear_alpha: 32
27
+ type: lora
28
+ save:
29
+ dtype: bf16
30
+ max_step_saves_to_keep: 4
31
+ save_every: 250
32
+ save_format: diffusers
33
+ train:
34
+ batch_size: 4
35
+ dtype: bf16
36
+ ema_config:
37
+ ema_decay: 0.99
38
+ use_ema: true
39
+ gradient_checkpointing: true
40
+ lr: 0.0001
41
+ noise_scheduler: flowmatch
42
+ optimizer: adamw8bit
43
+ steps: 2000
44
+ timestep_type: weighted
45
+ training_folder: /app/checkpoints/1c93dd95-2e89-48d9-813d-e0f521599cfd/z-image
46
+ type: diffusion_trainer
47
+ job: extension
48
+ meta:
49
+ name: zimage_lora
50
+ version: '1.0'