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

helloyongyang's avatar
helloyongyang committed
3
# set path and first
PengGao's avatar
PengGao committed
4
5
lightx2v_path=/mnt/afs/users/lijiaqi2/deploy-comfyui-ljq-custom_nodes/ComfyUI-Lightx2vWrapper/lightx2v
model_path=/mnt/afs/users/lijiaqi2/wan_model/Wan2.1-R2V0909-Audio-14B-720P-fp8
Zhuguanyu Wu's avatar
Zhuguanyu Wu committed
6

PengGao's avatar
PengGao committed
7
8

export CUDA_VISIBLE_DEVICES=0,1,2,3
Zhuguanyu Wu's avatar
Zhuguanyu Wu committed
9

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

# Start multiple servers
PengGao's avatar
PengGao committed
14
15
torchrun --nproc_per_node 4 -m lightx2v.server \
    --model_cls seko_talk \
16
    --task i2v \
Zhuguanyu Wu's avatar
Zhuguanyu Wu committed
17
    --model_path $model_path \
PengGao's avatar
PengGao committed
18
19
    --config_json ${lightx2v_path}/configs/seko_talk/xxx_dist.json \
    --port 8000