mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-26 08:48:03 +00:00
Add StableAnimator @ 0f3d85ad217c0d3edec89e310bb34c3ecb9eaf9b
https://github.com/Francis-Rings/StableAnimator/commit/0f3d85ad217c0d3edec89e310bb34c3ecb9eaf9b
This commit is contained in:
@@ -0,0 +1,23 @@
|
||||
CUDA_VISIBLE_DEVICES=3,2,1,0 accelerate launch train_single.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" \
|
||||
--data_path="path/animation_data/video_path.txt" \
|
||||
--dataset_width=512 \
|
||||
--dataset_height=512 \
|
||||
--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"
|
||||
Reference in New Issue
Block a user