start_multi_servers.sh 459 Bytes
Newer Older
Zhuguanyu Wu's avatar
Zhuguanyu Wu committed
1
2
#!/bin/bash

helloyongyang's avatar
helloyongyang committed
3
# set path and first
Zhuguanyu Wu's avatar
Zhuguanyu Wu committed
4
5
6
lightx2v_path=
model_path=

helloyongyang's avatar
helloyongyang committed
7
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
Zhuguanyu Wu's avatar
Zhuguanyu Wu committed
8

helloyongyang's avatar
helloyongyang committed
9
10
# set environment variables
source ${lightx2v_path}/scripts/base/base.sh
Zhuguanyu Wu's avatar
Zhuguanyu Wu committed
11
12
13
14
15

# Start multiple servers
python -m lightx2v.api_multi_servers \
    --num_gpus $num_gpus \
    --start_port 8000 \
16
17
    --model_cls wan2.1_distill \
    --task i2v \
Zhuguanyu Wu's avatar
Zhuguanyu Wu committed
18
    --model_path $model_path \
19
    --config_json ${lightx2v_path}/configs/distill/wan_i2v_distill_4step_cfg.json