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

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

Browse files
checkpoints/config.yaml ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ config:
2
+ name: last
3
+ process:
4
+ - datasets:
5
+ - cache_latents_to_disk: true
6
+ caption_dropout_rate: 0.0
7
+ caption_ext: txt
8
+ folder_path: /dataset/images/1c93dd95-2e89-48d9-813d-e0f521599cfd/img
9
+ is_reg: false
10
+ resolution:
11
+ - 512
12
+ - 768
13
+ - 1024
14
+ device: cuda
15
+ model:
16
+ arch: zimage:turbo
17
+ assistant_lora_path: /cache/hf_cache/zimage_turbo_training_adapter_v2.safetensors
18
+ name_or_path: /cache/models/gradients-io-tournaments--Z-Image-Turbo
19
+ qtype: qfloat8
20
+ qtype_te: qfloat8
21
+ quantize: true
22
+ quantize_te: true
23
+ network:
24
+ conv: 16
25
+ conv_alpha: 16
26
+ linear: 32
27
+ linear_alpha: 32
28
+ type: lora
29
+ save:
30
+ dtype: bf16
31
+ max_step_saves_to_keep: 4
32
+ save_every: 50
33
+ save_format: diffusers
34
+ train:
35
+ batch_size: 4
36
+ dtype: bf16
37
+ ema_config:
38
+ ema_decay: 0.99
39
+ use_ema: true
40
+ gradient_accumulation_steps: 1
41
+ gradient_checkpointing: true
42
+ loss_type: l2
43
+ lr: 0.0001
44
+ lr_scheduler: cosine
45
+ lr_scheduler_args:
46
+ num_cycles: 1
47
+ power: 1.0
48
+ lr_warmup_steps: 25
49
+ max_train_epochs: 1
50
+ noise_scheduler: flowmatch
51
+ optimizer: adamw8bit
52
+ optimizer_args:
53
+ betas:
54
+ - 0.9
55
+ - 0.999
56
+ eps: 1.0e-08
57
+ weight_decay: 0.0001
58
+ timestep_type: weighted
59
+ training_folder: /app/checkpoints/1c93dd95-2e89-48d9-813d-e0f521599cfd/z-image
60
+ type: diffusion_trainer
61
+ job: extension
62
+ meta:
63
+ name: zimage_lora
64
+ version: '1.0'
checkpoints/last_000000050.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:41c990b4c30ba42160e0bd699ba53a3173a470fea81122febd86223e8a8d594c
3
+ size 170128208
checkpoints/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d9f5610a40afdb9743947924a0aed3bca4c9be2cb85ac4520895ee00e1faa10a
3
+ size 173287845