Commit 4f135263 authored by Zhuguanyu Wu's avatar Zhuguanyu Wu Committed by GitHub
Browse files

Dev scheduler (#128)

* use CM scheduler for distill_models as default

* update configs
parent 6af19588
...@@ -11,5 +11,5 @@ ...@@ -11,5 +11,5 @@
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
"cpu_offload": false, "cpu_offload": false,
"denoising_step_list": [999, 750, 500, 250] "denoising_step_list": [1000, 750, 500, 250]
} }
...@@ -11,7 +11,7 @@ ...@@ -11,7 +11,7 @@
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
"cpu_offload": false, "cpu_offload": false,
"denoising_step_list": [999, 750, 500, 250], "denoising_step_list": [1000, 750, 500, 250],
"lora_path": [ "lora_path": [
"Wan2.1-I2V-14B-480P/loras/Wan21_I2V_14B_lightx2v_cfg_step_distill_lora_rank64.safetensors" "Wan2.1-I2V-14B-480P/loras/Wan21_I2V_14B_lightx2v_cfg_step_distill_lora_rank64.safetensors"
] ]
......
...@@ -9,8 +9,8 @@ ...@@ -9,8 +9,8 @@
"cross_attn_2_type": "flash_attn3", "cross_attn_2_type": "flash_attn3",
"seed": 42, "seed": 42,
"sample_guide_scale": 6, "sample_guide_scale": 6,
"sample_shift": 8, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
"cpu_offload": false, "cpu_offload": false,
"denoising_step_list": [999, 750, 500, 250] "denoising_step_list": [1000, 750, 500, 250]
} }
...@@ -9,10 +9,10 @@ ...@@ -9,10 +9,10 @@
"cross_attn_2_type": "flash_attn3", "cross_attn_2_type": "flash_attn3",
"seed": 42, "seed": 42,
"sample_guide_scale": 6, "sample_guide_scale": 6,
"sample_shift": 8, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
"enable_dynamic_cfg": true, "enable_dynamic_cfg": true,
"cfg_scale": 4.0, "cfg_scale": 4.0,
"cpu_offload": false, "cpu_offload": false,
"denoising_step_list": [999, 750, 500, 250] "denoising_step_list": [1000, 750, 500, 250]
} }
...@@ -9,11 +9,11 @@ ...@@ -9,11 +9,11 @@
"cross_attn_2_type": "flash_attn3", "cross_attn_2_type": "flash_attn3",
"seed": 42, "seed": 42,
"sample_guide_scale": 6, "sample_guide_scale": 6,
"sample_shift": 8, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
"cpu_offload": false, "cpu_offload": false,
"denoising_step_list": [999, 750, 500, 250], "denoising_step_list": [1000, 750, 500, 250],
"lora_path": [ "lora_path": [
"Wan2.1-T2V-14B/loras/Wan21_T2V_14B_lightx2v_cfg_step_distill_lora_rank32.safetensors" "Wan2.1-T2V-14B/loras/Wan21_T2V_14B_lightx2v_cfg_step_distill_lora_rank64.safetensors"
] ]
} }
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment