Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
xuwx1
LightX2V
Commits
e7e74dcd
"csrc/vscode:/vscode.git/clone" did not exist on "3b6c702f1e2dcf87bdf7ee52dbf452955e2fea6a"
Commit
e7e74dcd
authored
Aug 06, 2025
by
helloyongyang
Browse files
update wan2.2 dist scripts
parent
978e3b32
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
120 additions
and
0 deletions
+120
-0
configs/dist_infer/wan22_moe_t2v_cfg_ulysses.json
configs/dist_infer/wan22_moe_t2v_cfg_ulysses.json
+22
-0
configs/dist_infer/wan22_moe_t2v_ulysses.json
configs/dist_infer/wan22_moe_t2v_ulysses.json
+21
-0
scripts/dist_infer/run_wan22_moe_t2v_cfg_ulysses.sh
scripts/dist_infer/run_wan22_moe_t2v_cfg_ulysses.sh
+39
-0
scripts/dist_infer/run_wan22_moe_t2v_ulysses.sh
scripts/dist_infer/run_wan22_moe_t2v_ulysses.sh
+38
-0
No files found.
configs/dist_infer/wan22_moe_t2v_cfg_ulysses.json
0 → 100755
View file @
e7e74dcd
{
"infer_steps"
:
40
,
"target_video_length"
:
81
,
"text_len"
:
512
,
"target_height"
:
720
,
"target_width"
:
1280
,
"self_attn_1_type"
:
"flash_attn3"
,
"cross_attn_1_type"
:
"flash_attn3"
,
"cross_attn_2_type"
:
"flash_attn3"
,
"seed"
:
42
,
"sample_guide_scale"
:
[
4.0
,
3.0
],
"sample_shift"
:
12.0
,
"enable_cfg"
:
true
,
"cpu_offload"
:
true
,
"offload_granularity"
:
"model"
,
"boundary"
:
0.875
,
"parallel"
:
{
"seq_p_size"
:
4
,
"seq_p_attn_type"
:
"ulysses"
,
"cfg_p_size"
:
2
}
}
configs/dist_infer/wan22_moe_t2v_ulysses.json
0 → 100755
View file @
e7e74dcd
{
"infer_steps"
:
40
,
"target_video_length"
:
81
,
"text_len"
:
512
,
"target_height"
:
720
,
"target_width"
:
1280
,
"self_attn_1_type"
:
"flash_attn3"
,
"cross_attn_1_type"
:
"flash_attn3"
,
"cross_attn_2_type"
:
"flash_attn3"
,
"seed"
:
42
,
"sample_guide_scale"
:
[
4.0
,
3.0
],
"sample_shift"
:
12.0
,
"enable_cfg"
:
true
,
"cpu_offload"
:
true
,
"offload_granularity"
:
"model"
,
"boundary"
:
0.875
,
"parallel"
:
{
"seq_p_size"
:
4
,
"seq_p_attn_type"
:
"ulysses"
}
}
scripts/dist_infer/run_wan22_moe_t2v_cfg_ulysses.sh
0 → 100755
View file @
e7e74dcd
#!/bin/bash
# set path and first
lightx2v_path
=
model_path
=
# check section
if
[
-z
"
${
CUDA_VISIBLE_DEVICES
}
"
]
;
then
cuda_devices
=
0,1,2,3,4,5,6,7
echo
"Warn: CUDA_VISIBLE_DEVICES is not set, using default value:
${
cuda_devices
}
, change at shell script or set env variable."
export
CUDA_VISIBLE_DEVICES
=
${
cuda_devices
}
fi
if
[
-z
"
${
lightx2v_path
}
"
]
;
then
echo
"Error: lightx2v_path is not set. Please set this variable first."
exit
1
fi
if
[
-z
"
${
model_path
}
"
]
;
then
echo
"Error: model_path is not set. Please set this variable first."
exit
1
fi
export
TOKENIZERS_PARALLELISM
=
false
export
PYTHONPATH
=
${
lightx2v_path
}
:
$PYTHONPATH
export
DTYPE
=
BF16
export
SENSITIVE_LAYER_DTYPE
=
FP32
export
ENABLE_PROFILING_DEBUG
=
true
export
ENABLE_GRAPH_MODE
=
false
torchrun
--nproc_per_node
=
8
-m
lightx2v.infer
\
--model_cls
wan2.2_moe
\
--task
t2v
\
--model_path
$model_path
\
--config_json
${
lightx2v_path
}
/configs/dist_infer/wan22_moe_t2v_cfg_ulysses.json
\
--prompt
"Two anthropomorphic cats in comfy boxing gear and bright gloves fight intensely on a spotlighted stage."
\
--negative_prompt
"色调艳丽,过曝,静态,细节模糊不清,字幕,风格,作品,画作,画面,静止,整体发灰,最差质量,低质量,JPEG压缩残留,丑陋的,残缺的,多余的手指,画得不好的手部,画得不好的脸部,畸形的,毁容的,形态畸形的肢体,手指融合,静止不动的画面,杂乱的背景,三条腿,背景人很多,倒着走"
\
--save_video_path
${
lightx2v_path
}
/save_results/output_lightx2v_wan22_moe_t2v_parallel_cfg_ulysses.mp4
scripts/dist_infer/run_wan22_moe_t2v_ulysses.sh
0 → 100755
View file @
e7e74dcd
#!/bin/bash
# set path and first
lightx2v_path
=
model_path
=
# check section
if
[
-z
"
${
CUDA_VISIBLE_DEVICES
}
"
]
;
then
cuda_devices
=
0,1,2,3
echo
"Warn: CUDA_VISIBLE_DEVICES is not set, using default value:
${
cuda_devices
}
, change at shell script or set env variable."
export
CUDA_VISIBLE_DEVICES
=
${
cuda_devices
}
fi
if
[
-z
"
${
lightx2v_path
}
"
]
;
then
echo
"Error: lightx2v_path is not set. Please set this variable first."
exit
1
fi
if
[
-z
"
${
model_path
}
"
]
;
then
echo
"Error: model_path is not set. Please set this variable first."
exit
1
fi
export
TOKENIZERS_PARALLELISM
=
false
export
PYTHONPATH
=
${
lightx2v_path
}
:
$PYTHONPATH
export
DTYPE
=
BF16
export
ENABLE_PROFILING_DEBUG
=
true
export
ENABLE_GRAPH_MODE
=
false
torchrun
--nproc_per_node
=
4
-m
lightx2v.infer
\
--model_cls
wan2.2_moe
\
--task
t2v
\
--model_path
$model_path
\
--config_json
${
lightx2v_path
}
/configs/dist_infer/wan22_moe_t2v_ulysses.json
\
--prompt
"Two anthropomorphic cats in comfy boxing gear and bright gloves fight intensely on a spotlighted stage."
\
--negative_prompt
"色调艳丽,过曝,静态,细节模糊不清,字幕,风格,作品,画作,画面,静止,整体发灰,最差质量,低质量,JPEG压缩残留,丑陋的,残缺的,多余的手指,画得不好的手部,画得不好的脸部,畸形的,毁容的,形态畸形的肢体,手指融合,静止不动的画面,杂乱的背景,三条腿,背景人很多,倒着走"
\
--save_video_path
${
lightx2v_path
}
/save_results/output_lightx2v_wan22_moe_t2v_parallel_ulysses.mp4
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment