Gege24 commited on
Commit
8e7b9ad
·
verified ·
1 Parent(s): 26e2377

Upload task output 710fbafa-2698-4896-abf5-b52f3d509ef5

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