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 @@
"sample_shift": 5,
"enable_cfg": false,
"cpu_offload": false,
"denoising_step_list": [999, 750, 500, 250]
"denoising_step_list": [1000, 750, 500, 250]
}
......@@ -11,7 +11,7 @@
"sample_shift": 5,
"enable_cfg": false,
"cpu_offload": false,
"denoising_step_list": [999, 750, 500, 250],
"denoising_step_list": [1000, 750, 500, 250],
"lora_path": [
"Wan2.1-I2V-14B-480P/loras/Wan21_I2V_14B_lightx2v_cfg_step_distill_lora_rank64.safetensors"
]
......
......@@ -9,8 +9,8 @@
"cross_attn_2_type": "flash_attn3",
"seed": 42,
"sample_guide_scale": 6,
"sample_shift": 8,
"sample_shift": 5,
"enable_cfg": false,
"cpu_offload": false,
"denoising_step_list": [999, 750, 500, 250]
"denoising_step_list": [1000, 750, 500, 250]
}
......@@ -9,10 +9,10 @@
"cross_attn_2_type": "flash_attn3",
"seed": 42,
"sample_guide_scale": 6,
"sample_shift": 8,
"sample_shift": 5,
"enable_cfg": false,
"enable_dynamic_cfg": true,
"cfg_scale": 4.0,
"cpu_offload": false,
"denoising_step_list": [999, 750, 500, 250]
"denoising_step_list": [1000, 750, 500, 250]
}
......@@ -9,11 +9,11 @@
"cross_attn_2_type": "flash_attn3",
"seed": 42,
"sample_guide_scale": 6,
"sample_shift": 8,
"sample_shift": 5,
"enable_cfg": false,
"cpu_offload": false,
"denoising_step_list": [999, 750, 500, 250],
"denoising_step_list": [1000, 750, 500, 250],
"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