command_train.sh

CUDA_VISIBLE_DEVICES=3,2,1,0 accelerate launch train.py \
 --pretrained_model_name_or_path="path/checkpoints/SVD/stable-video-diffusion-img2vid-xt" \
 --output_dir="path/checkpoints/Animation" \
 --data_root_path="path/animation_data" \
 --rec_data_path="path/animation_data/video_rec_path.txt" \
 --vec_data_path="path/animation_data/video_vec_path.txt" \
 --validation_image_folder="path/validation/ground_truth" \
 --validation_control_folder="path/validation/poses" \
 --validation_image="path/validation/reference.png" \
 --num_workers=8 \
 --lr_warmup_steps=500 \
 --sample_n_frames=16 \
 --learning_rate=1e-5 \
 --per_gpu_batch_size=1 \
 --num_train_epochs=6000 \
 --mixed_precision="fp16" \
 --gradient_accumulation_steps=1 \
 --checkpointing_steps=2000 \
 --validation_steps=500 \
 --gradient_checkpointing \
 --checkpoints_total_limit=5000 \
 --resume_from_checkpoint="latest"