run_gradio.sh 493 Bytes
Newer Older
gushiqiao's avatar
gushiqiao committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#!/bin/bash

lightx2v_path=/path/to/lightx2v
model_path=/path/to/wan

export CUDA_VISIBLE_DEVICES=0
export CUDA_LAUNCH_BLOCKING=1
export PYTHONPATH=${lightx2v_path}:$PYTHONPATH
export ENABLE_PROFILING_DEBUG=true
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True

python gradio_demo.py \
    --model_path $model_path \
    --server_name 0.0.0.0 \
    --server_port 8005

# python gradio_demo_zh.py \
#     --model_path $model_path \
#     --server_name 0.0.0.0 \
#     --server_port 8005