Unverified Commit 04812de2 authored by Yang Yong (雍洋)'s avatar Yang Yong (雍洋) Committed by GitHub
Browse files

Refactor Config System (#338)

parent 6a658f42
...@@ -6,7 +6,6 @@ ...@@ -6,7 +6,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 5, "sample_guide_scale": 5,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": true, "enable_cfg": true,
......
...@@ -7,7 +7,6 @@ ...@@ -7,7 +7,6 @@
"self_attn_1_type": "flash_attn3", "self_attn_1_type": "flash_attn3",
"cross_attn_1_type": "flash_attn3", "cross_attn_1_type": "flash_attn3",
"cross_attn_2_type": "flash_attn3", "cross_attn_2_type": "flash_attn3",
"seed": 42,
"sample_guide_scale": 6, "sample_guide_scale": 6,
"sample_shift": 8, "sample_shift": 8,
"enable_cfg": true, "enable_cfg": true,
......
...@@ -6,7 +6,6 @@ ...@@ -6,7 +6,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 5, "sample_guide_scale": 5,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": true, "enable_cfg": true,
......
...@@ -6,7 +6,6 @@ ...@@ -6,7 +6,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 5, "sample_guide_scale": 5,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
......
...@@ -6,7 +6,6 @@ ...@@ -6,7 +6,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 5, "sample_guide_scale": 5,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": true, "enable_cfg": true,
...@@ -14,9 +13,5 @@ ...@@ -14,9 +13,5 @@
"offload_granularity": "phase", "offload_granularity": "phase",
"t5_cpu_offload": true, "t5_cpu_offload": true,
"t5_offload_granularity": "block", "t5_offload_granularity": "block",
"dit_quantized_ckpt": "/path/to/dit_int8",
"mm_config": {
"mm_type": "W-int8-channel-sym-A-int8-channel-sym-dynamic-Q8F"
},
"use_tiling_vae": true "use_tiling_vae": true
} }
...@@ -7,7 +7,6 @@ ...@@ -7,7 +7,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 6, "sample_guide_scale": 6,
"sample_shift": 8, "sample_shift": 8,
"enable_cfg": true, "enable_cfg": true,
......
...@@ -3,7 +3,6 @@ ...@@ -3,7 +3,6 @@
"target_video_length": 33, "target_video_length": 33,
"i2v_resolution": "720p", "i2v_resolution": "720p",
"attention_type": "flash_attn3", "attention_type": "flash_attn3",
"seed": 0,
"dit_quantized_ckpt": "/path/to/int8/model", "dit_quantized_ckpt": "/path/to/int8/model",
"mm_config": { "mm_config": {
"mm_type": "W-int8-channel-sym-A-int8-channel-sym-dynamic-Vllm" "mm_type": "W-int8-channel-sym-A-int8-channel-sym-dynamic-Vllm"
......
...@@ -6,7 +6,6 @@ ...@@ -6,7 +6,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 5, "sample_guide_scale": 5,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": true, "enable_cfg": true,
......
...@@ -6,7 +6,6 @@ ...@@ -6,7 +6,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 5, "sample_guide_scale": 5,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
......
...@@ -6,7 +6,6 @@ ...@@ -6,7 +6,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 5, "sample_guide_scale": 5,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
......
{ {
"seed": 42,
"batchsize": 1, "batchsize": 1,
"num_channels_latents": 16, "num_channels_latents": 16,
"vae_scale_factor": 8, "vae_scale_factor": 8,
...@@ -23,7 +22,7 @@ ...@@ -23,7 +22,7 @@
-0.9497, -0.9497,
0.2503, 0.2503,
-0.2921 -0.2921
], ],
"vae_latents_std": [ "vae_latents_std": [
2.8184, 2.8184,
1.4541, 1.4541,
......
{ {
"seed": 42,
"batchsize": 1, "batchsize": 1,
"_comment": "格式: '宽高比': [width, height]", "_comment": "格式: '宽高比': [width, height]",
"aspect_ratios": { "aspect_ratios": {
"1:1": [1328, 1328], "1:1": [
"16:9": [1664, 928], 1328,
"9:16": [928, 1664], 1328
"4:3": [1472, 1140], ],
"3:4": [142, 184] "16:9": [
1664,
928
],
"9:16": [
928,
1664
],
"4:3": [
1472,
1140
],
"3:4": [
142,
184
]
}, },
"aspect_ratio": "16:9", "aspect_ratio": "16:9",
"num_channels_latents": 16, "num_channels_latents": 16,
...@@ -32,7 +46,7 @@ ...@@ -32,7 +46,7 @@
-0.9497, -0.9497,
0.2503, 0.2503,
-0.2921 -0.2921
], ],
"vae_latents_std": [ "vae_latents_std": [
2.8184, 2.8184,
1.4541, 1.4541,
......
...@@ -8,7 +8,6 @@ ...@@ -8,7 +8,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 1.0, "sample_guide_scale": 1.0,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
......
...@@ -8,7 +8,6 @@ ...@@ -8,7 +8,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 1.0, "sample_guide_scale": 1.0,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
......
...@@ -8,7 +8,6 @@ ...@@ -8,7 +8,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 1.0, "sample_guide_scale": 1.0,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
......
...@@ -8,7 +8,6 @@ ...@@ -8,7 +8,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 1.0, "sample_guide_scale": 1.0,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
......
...@@ -8,7 +8,6 @@ ...@@ -8,7 +8,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 1.0, "sample_guide_scale": 1.0,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
......
...@@ -8,7 +8,6 @@ ...@@ -8,7 +8,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 1.0, "sample_guide_scale": 1.0,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
......
...@@ -8,7 +8,6 @@ ...@@ -8,7 +8,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 1.0, "sample_guide_scale": 1.0,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
......
...@@ -8,7 +8,6 @@ ...@@ -8,7 +8,6 @@
"self_attn_1_type": "sage_attn2", "self_attn_1_type": "sage_attn2",
"cross_attn_1_type": "sage_attn2", "cross_attn_1_type": "sage_attn2",
"cross_attn_2_type": "sage_attn2", "cross_attn_2_type": "sage_attn2",
"seed": 42,
"sample_guide_scale": 1.0, "sample_guide_scale": 1.0,
"sample_shift": 5, "sample_shift": 5,
"enable_cfg": false, "enable_cfg": false,
......
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