mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-09 00:09:55 +00:00
Brandon Thomas
f912a79914
https://github.com/Francis-Rings/StableAnimator/commit/0f3d85ad217c0d3edec89e310bb34c3ecb9eaf9b
22 lines
886 B
Bash
22 lines
886 B
Bash
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" |