update scripts

scripts/wan/run_wan_i2v.sh

@@ -4,30 +4,10 @@
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-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 ENABLE_PROFILING_DEBUG=true
-export ENABLE_GRAPH_MODE=false
-export DTYPE=BF16  # remove this can get high quality video
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.1 \

scripts/wan/run_wan_i2v_audio.sh

 #!/bin/bash
 
 # set path and first
-
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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 ENABLE_PROFILING_DEBUG=true
-export ENABLE_GRAPH_MODE=false
-export DTYPE=BF16
+export CUDA_VISIBLE_DEVICES=0
 
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 export TORCH_CUDA_ARCH_LIST="9.0"
 export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True

scripts/wan/run_wan_i2v_causvid.sh

@@ -3,29 +3,11 @@
 # set path and first
 lightx2v_path=
 model_path=
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-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
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.1_causvid \

scripts/wan/run_wan_i2v_distill_4step_cfg.sh

@@ -4,30 +4,10 @@
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-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 ENABLE_PROFILING_DEBUG=true
-export ENABLE_GRAPH_MODE=false
-export DTYPE=BF16
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.1_distill \

scripts/wan/run_wan_i2v_distill_4step_cfg_lora.sh

@@ -4,30 +4,10 @@
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-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 ENABLE_PROFILING_DEBUG=true
-export ENABLE_GRAPH_MODE=false
-export DTYPE=BF16
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.1_distill \

scripts/wan/run_wan_i2v_lazy_load.sh

 #!/bin/bash
 
 # set path and first
-lightx2v_path=/data/video_gen/lightx2v_latest/lightx2v
-model_path=/data/video_gen/x2v_models/Wan2.1-I2V-14B-720P
+lightx2v_path=
+model_path=
 
+export CUDA_VISIBLE_DEVICES=0
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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 PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
-export PYTHONPATH=${lightx2v_path}:$PYTHONPATH
-
-export ENABLE_PROFILING_DEBUG=true
-export ENABLE_GRAPH_MODE=true
-export DTYPE=BF16  # remove this can get high quality video
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.1 \

scripts/wan/run_wan_skyreels_v2_df.sh

 #!/bin/bash
+
 # set path and first
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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 CUDA_VISIBLE_DEVICES=0
 
-export PYTHONPATH=${lightx2v_path}:$PYTHONPATH
-export DTYPE=BF16
-export ENABLE_PROFILING_DEBUG=true
-export ENABLE_GRAPH_MODE=false
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 #I2V
 python -m lightx2v.infer \

scripts/wan/run_wan_skyreels_v2_i2v.sh

 #!/bin/bash
+
 # set path and first
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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 CUDA_VISIBLE_DEVICES=0
 
-export PYTHONPATH=${lightx2v_path}:$PYTHONPATH
-export DTYPE=BF16
-export ENABLE_PROFILING_DEBUG=true
-export ENABLE_GRAPH_MODE=false
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.1 \

scripts/wan/run_wan_skyreels_v2_t2v.sh

@@ -4,29 +4,10 @@
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-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
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.1 \

scripts/wan/run_wan_t2v.sh

@@ -4,29 +4,10 @@
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-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
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.1 \

scripts/wan/run_wan_t2v_causvid.sh

@@ -4,29 +4,10 @@
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-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
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.1_causvid \

scripts/wan/run_wan_t2v_distill_4step_cfg.sh

@@ -4,29 +4,10 @@
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-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
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.1_distill \

scripts/wan/run_wan_t2v_distill_4step_cfg_dynamic.sh

@@ -4,29 +4,10 @@
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-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
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.1_distill \

scripts/wan/run_wan_t2v_distill_4step_cfg_lora.sh

@@ -4,29 +4,10 @@
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-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
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.1_distill \

scripts/wan/run_wan_t2v_enhancer.sh

-#!/bin/bash
-
-# set path and first
-lightx2v_path=
-model_path=
-prompt_enhancer_path=
-
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0,1
-    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 ENABLE_PROFILING_DEBUG=true
-export ENABLE_GRAPH_MODE=false
-export DTYPE=BF16
-
-python -m lightx2v.infer \
---model_cls wan2.1 \
---task t2v \
---model_path $model_path \
---config_json ${lightx2v_path}/configs/wan/wan_t2v_enhancer.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_wan_t2v.mp4 \
---use_prompt_enhancer

scripts/wan22/run_wan22_moe_i2v.sh

@@ -4,29 +4,10 @@
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-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
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.2_moe \

scripts/wan22/run_wan22_moe_i2v_audio.sh

@@ -4,36 +4,15 @@
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-if [ -z "${lightx2v_path}" ]; then
-    echo "Error: lightx2v_path is not set. Please set this variable first."
-    exit 1
-fi
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
-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
 export TORCH_CUDA_ARCH_LIST="9.0"
 export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
 
-#-m debugpy --wait-for-client --listen 0.0.0.0:15684 \
-
-python \
--m lightx2v.infer \
+python -m lightx2v.infer \
 --model_cls wan2.2_moe_audio \
 --task i2v \
 --model_path $model_path \

scripts/wan22/run_wan22_moe_t2v.sh

@@ -4,29 +4,10 @@
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-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
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.2_moe \

scripts/wan22/run_wan22_moe_t2v_distill.sh

@@ -4,29 +4,10 @@
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-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
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.2_moe_distill \

scripts/wan22/run_wan22_ti2v_i2v.sh

@@ -4,29 +4,10 @@
 lightx2v_path=
 model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=5
-    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
+export CUDA_VISIBLE_DEVICES=0
 
-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
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
 --model_cls wan2.2 \