mirror of
https://github.com/THUDM/CogVideo.git
synced 2025-04-06 03:57:56 +08:00
10 lines
280 B
Bash
10 lines
280 B
Bash
#! /bin/bash
|
|
|
|
echo "RUN on $(hostname), CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES"
|
|
|
|
run_cmd="torchrun --standalone --nproc_per_node=8 train_video.py --base configs/cogvideox_2b_lora.yaml configs/sft.yaml --seed $RANDOM"
|
|
|
|
echo ${run_cmd}
|
|
eval ${run_cmd}
|
|
|
|
echo "DONE on `hostname`" |