Skip to content

GitLab

Commit dc296c2f authored by helloyongyang's avatar helloyongyang
Browse files

update audio scripts

parent 95b58beb
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
scripts/wan/run_wan_i2v_audio.sh
View file @ dc296c2f
...@@ -12,6 +12,9 @@ source ${lightx2v_path}/scripts/base/base.sh ...@@ -12,6 +12,9 @@ source ${lightx2v_path}/scripts/base/base.sh
export TORCH_CUDA_ARCH_LIST="9.0" export TORCH_CUDA_ARCH_LIST="9.0"
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
export ENABLE_GRAPH_MODE=false
export SENSITIVE_LAYER_DTYPE=None
python -m lightx2v.infer \ python -m lightx2v.infer \
--model_cls wan2.1_audio \ --model_cls wan2.1_audio \
--task i2v \ --task i2v \
......
scripts/wan/run_wan_i2v_audio_dist.sh
View file @ dc296c2f
...@@ -13,6 +13,7 @@ export TORCH_CUDA_ARCH_LIST="9.0" ...@@ -13,6 +13,7 @@ export TORCH_CUDA_ARCH_LIST="9.0"
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
export ENABLE_GRAPH_MODE=false export ENABLE_GRAPH_MODE=false
export SENSITIVE_LAYER_DTYPE=None
#for debugging #for debugging
#export TORCH_NCCL_BLOCKING_WAIT=1 #启用 NCCL 阻塞等待模式（否则 watchdog 会杀死卡顿的进程） #export TORCH_NCCL_BLOCKING_WAIT=1 #启用 NCCL 阻塞等待模式（否则 watchdog 会杀死卡顿的进程）
......
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