Skip to content

GitLab

Commit 8bc0da34 authored by gushiqiao's avatar gushiqiao Committed by GitHub
Browse files

reconstruct config and scripts (#81) 


Co-authored-by: default avatargushiqiao <gushiqiao@sensetime.com>
parent ad051778
Hide whitespace changes
Inline Side-by-side
Showing with 171 additions and 7 deletions
configs/caching/wan_i2v_tea.json → configs/attns/wan_i2v_flash.json
View file @ 8bc0da34
......@@ -4,10 +4,9 @@
"target_height": 480,
"target_width": 832,
"attention_type": "flash_attn3",
"seed": 442,
"seed": 42,
"sample_guide_scale": 5,
"sample_shift": 3,
"sample_shift": 5,
"enable_cfg": true,
"cpu_offload": false,
"feature_caching": "Tea"
"cpu_offload": false
}
configs/offload/wan_i2v_block.json → configs/attns/wan_i2v_sage.json
View file @ 8bc0da34
......@@ -8,6 +8,5 @@
"sample_guide_scale": 5,
"sample_shift": 5,
"enable_cfg": true,
"cpu_offload": true,
"offload_granularity": "block"
"cpu_offload": false
}
configs/wan_t2v_sparge.json → configs/attns/wan_t2v_sparge.json
View file @ 8bc0da34
......@@ -11,5 +11,5 @@
"enable_cfg": true,
"cpu_offload": false,
"sparge": true,
"sparge_ckpt": "configs/shared_weights/sparge_wan2.1_t2v_1.3B.pt"
"sparge_ckpt": "/path/to/sparge_wan2.1_t2v_1.3B.pt"
}
configs/caching/wan_i2v_custom.json → configs/caching/custom/wan_i2v_custom_480p.json
View file @ 8bc0da34
......@@ -9,5 +9,11 @@
"sample_shift": 3,
"enable_cfg": true,
"cpu_offload": false,
"feature_caching": "Custom"
"feature_caching": "Custom",
"coefficients": [
[2.57151496e05, -3.54229917e04, 1.40286849e03, -1.35890334e01, 1.32517977e-01],
[-3.02331670e02, 2.23948934e02, -5.25463970e01, 5.87348440e00, -2.01973289e-01]
],
"use_ret_steps": true,
"teacache_thresh": 0.26
}
configs/caching/custom/wan_i2v_custom_720p.json 0 → 100755
View file @ 8bc0da34
{
"infer_steps": 40,
"target_video_length": 81,
"target_height": 1280,
"target_width": 720,
"attention_type": "flash_attn3",
"seed": 442,
"sample_guide_scale": 5,
"sample_shift": 3,
"enable_cfg": true,
"cpu_offload": false,
"feature_caching": "Custom",
"coefficients": [
[8.10705460e03, 2.13393892e03, -3.72934672e02, 1.66203073e01, -4.17769401e-02],
[-114.36346466, 65.26524496, -18.82220707, 4.91518089, -0.23412683]
],
"use_ret_steps": true,
"teacache_thresh": 0.26
}
configs/caching/wan_t2v_custom.json → configs/caching/custom/wan_t2v_custom_14b.json
View file @ 8bc0da34
......@@ -10,5 +10,11 @@
"sample_shift": 8,
"enable_cfg": true,
"cpu_offload": false,
"feature_caching": "Custom"
"feature_caching": "Custom",
"coefficients": [
[-3.03318725e05, 4.90537029e04, -2.65530556e03, 5.87365115e01, -3.15583525e-01],
[-5784.54975374, 5449.50911966, -1811.16591783, 256.27178429, -13.02252404]
],
"use_ret_steps": true,
"teacache_thresh": 0.26
}
configs/caching/custom/wan_t2v_custom_1_3b.json 0 → 100755
View file @ 8bc0da34
{
"infer_steps": 50,
"target_video_length": 81,
"text_len": 512,
"target_height": 480,
"target_width": 832,
"attention_type": "flash_attn3",
"seed": 42,
"sample_guide_scale": 6,
"sample_shift": 8,
"enable_cfg": true,
"cpu_offload": false,
"feature_caching": "Custom",
"coefficients": [
[-5.21862437e04, 9.23041404e03, -5.28275948e02, 1.36987616e01, -4.99875664e-02],
[2.39676752e03, -1.31110545e03, 2.01331979e02, -8.29855975e00, 1.37887774e-01]
],
"use_ret_steps": true,
"teacache_thresh": 0.26
}
configs/caching/teacache/wan_i2v_tea_480p.json 0 → 100755
View file @ 8bc0da34
{
"infer_steps": 40,
"target_video_length": 81,
"target_height": 480,
"target_width": 832,
"attention_type": "flash_attn3",
"seed": 42,
"sample_guide_scale": 5,
"sample_shift": 5,
"enable_cfg": true,
"cpu_offload": false,
"feature_caching": "Tea",
"coefficients": [
[2.57151496e05, -3.54229917e04, 1.40286849e03, -1.35890334e01, 1.32517977e-01],
[-3.02331670e02, 2.23948934e02, -5.25463970e01, 5.87348440e00, -2.01973289e-01]
],
"use_ret_steps": true,
"teacache_thresh": 0.26
}
configs/caching/teacache/wan_i2v_tea_720p.json 0 → 100755
View file @ 8bc0da34
{
"infer_steps": 40,
"target_video_length": 81,
"target_height": 1280,
"target_width": 720,
"attention_type": "flash_attn3",
"seed": 42,
"sample_guide_scale": 5,
"sample_shift": 5,
"enable_cfg": true,
"cpu_offload": false,
"feature_caching": "Tea",
"coefficients": [
[8.10705460e03, 2.13393892e03, -3.72934672e02, 1.66203073e01, -4.17769401e-02],
[-114.36346466, 65.26524496, -18.82220707, 4.91518089, -0.23412683]
],
"use_ret_steps": true,
"teacache_thresh": 0.26
}
configs/caching/teacache/wan_t2v_14b.json 0 → 100755
View file @ 8bc0da34
{
"infer_steps": 50,
"target_video_length": 81,
"text_len": 512,
"target_height": 480,
"target_width": 832,
"attention_type": "flash_attn3",
"seed": 42,
"sample_guide_scale": 6,
"sample_shift": 8,
"enable_cfg": true,
"cpu_offload": false,
"mm_config": {
"mm_type": "W-fp8-channel-sym-A-fp8-channel-sym-dynamic-Sgl",
"weight_auto_quant": true
},
"feature_caching": "Tea",
"coefficients": [
[-3.03318725e05, 4.90537029e04, -2.65530556e03, 5.87365115e01, -3.15583525e-01],
[-5784.54975374, 5449.50911966, -1811.16591783, 256.27178429, -13.02252404]
],
"use_ret_steps": true,
"teacache_thresh": 0.26
}
configs/caching/wan_t2v_tea.json → configs/caching/teacache/wan_t2v_1_3b.json
View file @ 8bc0da34
......@@ -10,5 +10,11 @@
"sample_shift": 8,
"enable_cfg": true,
"cpu_offload": false,
"feature_caching": "Tea"
"feature_caching": "Tea",
"coefficients": [
[-5.21862437e04, 9.23041404e03, -5.28275948e02, 1.36987616e01, -4.99875664e-02],
[2.39676752e03, -1.31110545e03, 2.01331979e02, -8.29855975e00, 1.37887774e-01]
],
"use_ret_steps": true,
"teacache_thresh": 0.26
}
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