mirror of
https://github.com/THUDM/CogVideo.git
synced 2025-04-05 19:41:59 +08:00
12 lines
312 B
Bash
Executable File
12 lines
312 B
Bash
Executable File
#! /bin/bash
|
|
|
|
echo "CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES"
|
|
|
|
environs="WORLD_SIZE=1 RANK=0 LOCAL_RANK=0 LOCAL_WORLD_SIZE=1"
|
|
|
|
run_cmd="$environs python sample_video.py --base configs/test_cogvideox_5b.yaml configs/test_inference.yaml --seed $RANDOM"
|
|
|
|
echo ${run_cmd}
|
|
eval ${run_cmd}
|
|
|
|
echo "DONE on `hostname`" |