Jordansky commited on
Commit
75ac448
·
verified ·
1 Parent(s): 3393d84

Upload task output 925ea150-6c47-4472-8380-178eabc42cba

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/925ea150-6c47-4472-8380-178eabc42cba/img
8
+ is_reg: false
9
+ resolution:
10
+ - 512
11
+ - 768
12
+ - 1024
13
+ device: cuda
14
+ model:
15
+ arch: qwen_image
16
+ low_vram: true
17
+ name_or_path: /cache/models/gradients-io-tournaments--Qwen-Image
18
+ qtype: float8
19
+ quantize: true
20
+ quantize_te: true
21
+ network:
22
+ linear: 64
23
+ linear_alpha: 128
24
+ type: lora
25
+ save:
26
+ dtype: bf16
27
+ max_step_saves_to_keep: 2
28
+ save_every: 250
29
+ save_format: diffusers
30
+ train:
31
+ batch_size: 2
32
+ cache_text_embeddings: true
33
+ cfg_scale: 5.0
34
+ do_cfg: true
35
+ dtype: bf16
36
+ ema_config:
37
+ ema_decay: 0.995
38
+ use_ema: true
39
+ gradient_accumulation: 1
40
+ gradient_checkpointing: true
41
+ lr: 4.0e-05
42
+ noise_scheduler: flowmatch
43
+ optimizer: adamw8bit
44
+ optimizer_params:
45
+ weight_decay: 1.0e-05
46
+ steps: 2500
47
+ timestep_type: weighted
48
+ train_text_encoder: false
49
+ train_unet: true
50
+ training_folder: /app/checkpoints/925ea150-6c47-4472-8380-178eabc42cba/qwen2h-2
51
+ type: diffusion_trainer
52
+ job: extension
53
+ meta:
54
+ name: qwen_image_lora
55
+ version: '1.0'