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