rkps commited on
Commit
f73882a
·
verified ·
1 Parent(s): e48cf1f

Upload configs/train_low.toml with huggingface_hub

Browse files
Files changed (1) hide show
  1. configs/train_low.toml +54 -0
configs/train_low.toml ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Wan 2.2 I2V 14B - LOW NOISE LoRA training
2
+ # Low noise model handles details, anatomy, fine textures
3
+
4
+ output_dir = '/home/u/work/output_low'
5
+ dataset = '/home/u/work/diffusion-pipe/configs/dataset_blink.toml'
6
+
7
+ # training settings
8
+ epochs = 1000
9
+ max_steps = 20000
10
+ micro_batch_size_per_gpu = 2
11
+ pipeline_stages = 1
12
+ gradient_accumulation_steps = 1
13
+ gradient_clipping = 1.0
14
+ warmup_steps = 100
15
+
16
+ # eval settings
17
+ eval_every_n_epochs = 1
18
+ eval_before_first_step = true
19
+ eval_micro_batch_size_per_gpu = 1
20
+ eval_gradient_accumulation_steps = 1
21
+
22
+ # misc settings
23
+ save_every_n_steps = 500
24
+ checkpoint_every_n_minutes = 60
25
+ activation_checkpointing = true
26
+ compile = true
27
+ partition_method = 'parameters'
28
+ save_dtype = 'bfloat16'
29
+ caching_batch_size = 1
30
+ steps_per_print = 1
31
+ video_clip_mode = 'single_beginning'
32
+
33
+ [model]
34
+ type = 'wan'
35
+ ckpt_path = '/home/u/work/models/Wan2.2-I2V-A14B'
36
+ transformer_path = '/home/u/work/models/comfyui/split_files/diffusion_models/wan2.2_i2v_low_noise_14B_fp16.safetensors'
37
+ llm_path = '/home/u/work/models/comfyui/split_files/text_encoders/umt5_xxl_fp16.safetensors'
38
+ dtype = 'bfloat16'
39
+ transformer_dtype = 'float8'
40
+ timestep_sample_method = 'logit_normal'
41
+ # Full timestep range for low noise — helps with high->low switching at inference
42
+ # (advice from PENISLORA author: train low on 0-1 instead of restricted range)
43
+
44
+ [adapter]
45
+ type = 'lora'
46
+ rank = 16
47
+ dtype = 'bfloat16'
48
+
49
+ [optimizer]
50
+ type = 'adamw_optimi'
51
+ lr = 2e-5
52
+ betas = [0.9, 0.99]
53
+ weight_decay = 0.01
54
+ eps = 1e-8