Skip to content

Commit bdd5c08

Browse files
committed
Update Ascend NPU platform configs and scripts
1 parent 313f82f commit bdd5c08

30 files changed

Lines changed: 277 additions & 32 deletions
File renamed without changes.

configs/platforms/ascend_npu/dist_8/hunyuan_video_t2v_480p.json renamed to configs/platforms/ascend_npu/dist/hunyuan_video_t2v_480p.json

File renamed without changes.
Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
{
2+
"infer_steps": 50,
3+
"aspect_ratio": "16:9",
4+
"attn_type": "npu_flash_attn",
5+
"enable_cfg": true,
6+
"sample_guide_scale": 4.0,
7+
"enable_cfg_renorm": true,
8+
"cfg_renorm_min": 0.0,
9+
"axes_dims_rope": [
10+
16,
11+
56,
12+
56
13+
],
14+
"dit_quant_scheme": "Default",
15+
"rms_norm_type": "torch",
16+
"modulate_type": "torch",
17+
"rope_type": "torch",
18+
"layer_norm_type": "torch",
19+
"parallel": {
20+
"seq_p_size": 4,
21+
"seq_p_attn_type": "ulysses",
22+
"cfg_p_size": 2
23+
}
24+
}

configs/platforms/ascend_npu/dist_8/ltx2_3.json renamed to configs/platforms/ascend_npu/dist/ltx2_3.json

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,9 @@
1111
],
1212
"enable_cfg": true,
1313
"cpu_offload": false,
14+
"gemma_cpu_offload": true,
15+
"vae_cpu_offload": false,
16+
"audio_mel_cpu_offload": true,
1417
"offload_granularity": "model",
1518
"norm_modulate_backend": "torch",
1619
"modulate_type": "torch",
@@ -23,7 +26,7 @@
2326
"audio_mel_bins": 16,
2427
"double_precision_rope": false,
2528
"use_tiling_vae": false,
26-
"dit_original_ckpt": "/data/wushuo1/to5_models/LTX-2.3/ltx-2.3-22b-dev.safetensors",
29+
"dit_original_ckpt": "/data/wushuo1/models/LTX-2.3/ltx-2.3-22b-dev.safetensors",
2730
"caption_proj_before_connector": true,
2831
"cross_attention_adaln": true,
2932
"apply_gated_attention": true,
@@ -51,7 +54,6 @@
5154
}
5255
},
5356
"parallel": {
54-
"seq_p_size": 8,
55-
"seq_p_attn_type": "ulysses"
57+
"tensor_p_size": 8
5658
}
5759
}

configs/platforms/ascend_npu/dist_8/qwen_image_t2i_2512.json renamed to configs/platforms/ascend_npu/dist/qwen_image_t2i_2512.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
"rope_type": "torch",
1313
"layer_norm_type": "torch",
1414
"rms_norm_type": "torch",
15+
"vae_decode_parallel": true,
1516
"parallel": {
1617
"seq_p_size": 4,
1718
"seq_p_attn_type": "ulysses",

configs/platforms/ascend_npu/dist_8/wan_moe_t2v.json renamed to configs/platforms/ascend_npu/dist/wan_moe_t2v.json

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,8 @@
22
"infer_steps": 40,
33
"target_video_length": 81,
44
"text_len": 512,
5-
"target_height": 720,
6-
"target_width": 1280,
5+
"target_height": 480,
6+
"target_width": 832,
77
"self_attn_1_type": "npu_flash_attn",
88
"cross_attn_1_type": "npu_flash_attn",
99
"cross_attn_2_type": "npu_flash_attn",
@@ -13,10 +13,8 @@
1313
],
1414
"sample_shift": 12.0,
1515
"enable_cfg": true,
16-
"cpu_offload": false,
16+
"cpu_offload": true,
1717
"offload_granularity": "model",
18-
"t5_cpu_offload": false,
19-
"vae_cpu_offload": false,
2018
"boundary": 0.875,
2119
"modulate_type": "torch",
2220
"rope_type": "torch",
File renamed without changes.

configs/platforms/ascend_npu/dist_8/wan_t2v_sf.json renamed to configs/platforms/ascend_npu/dist/wan_t2v_sf.json

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
"sample_shift": 5.0,
1212
"enable_cfg": false,
1313
"cpu_offload": false,
14-
"dit_original_ckpt": "/data/wushuo1/to5_models/Self-Forcing/checkpoints/self_forcing_dmd.pt",
14+
"dit_original_ckpt": "/data/wushuo1/models/Self-Forcing/checkpoints/self_forcing_dmd.pt",
1515
"causal_rope_type": "torch",
1616
"modulate_type": "torch",
1717
"rope_type": "torch",
@@ -30,7 +30,8 @@
3030
"async_vae_decode": false
3131
},
3232
"parallel": {
33-
"seq_p_size": 8,
33+
"seq_p_size": 4,
34+
"cfg_p_size": 2,
3435
"seq_p_attn_type": "ulysses"
3536
}
3637
}

configs/platforms/ascend_npu/dist_8/z_image_turbo_t2i.json renamed to configs/platforms/ascend_npu/dist/z_image_turbo_t2i.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
"layer_norm_type": "torch",
1414
"rms_norm_type": "torch",
1515
"parallel": {
16-
"seq_p_size": 8,
16+
"seq_p_size": 2,
1717
"seq_p_attn_type": "ulysses"
1818
}
1919
}

configs/platforms/ascend_npu/single/ltx2_3.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
"audio_mel_bins":16,
2121
"double_precision_rope": false,
2222
"use_tiling_vae": false,
23-
"dit_original_ckpt": "/data/wushuo1/to5_models/LTX-2.3/ltx-2.3-22b-dev.safetensors",
23+
"dit_original_ckpt": "/data/wushuo1/models/LTX-2.3/ltx-2.3-22b-dev.safetensors",
2424
"caption_proj_before_connector": true,
2525
"cross_attention_adaln": true,
2626
"apply_gated_attention": true,

0 commit comments

Comments
 (0)