reference script

animation/StableAnimator/command_advanced_infer.sh

@@ -0,0 +1,36 @@
+#!/bin/bash
+
+OUTPUT_DIR="inference/output_2/final_output"
+
+POSE_FRAME_DIR="inference/output_2/poses"
+STARTING_IMAGE="inference/output_1/frames/frame_0.png"
+
+#mkdir -p $OUTPUT_DIR
+
+#CUDA_VISIBLE_DEVICES=0 
+python inference_advanced.py \
+ --start_image_path="${STARTING_IMAGE}" \
+ --output_dir="${OUTPUT_DIR}" \
+ --pose_images_folder="/tmp/pose" \
+ --pre_pose_video_path="inference/zeihan_trimmed.mp4" \
+ --width=1024 \
+ --height=576 \
+ --pretrained_model_name_or_path="checkpoints/stable-video-diffusion-img2vid-xt" \
+ --posenet_model_name_or_path="checkpoints/Animation/pose_net.pth" \
+ --face_encoder_model_name_or_path="checkpoints/Animation/face_encoder.pth" \
+ --unet_model_name_or_path="checkpoints/Animation/unet.pth" \
+ --guidance_scale=3.0 \
+ --num_inference_steps=25 \
+ --tile_size=16 \
+ --overlap=4 \
+ --noise_aug_strength=0.02 \
+ --frames_overlap=4 \
+ --decode_chunk_size=4 \
+ --gradient_checkpointing
+
+
+#ffmpeg -framerate 30 \
+#  -start_number 0 \
+#  -i "${OUTPUT_DIR}/animated_images/frame_%d.png" \
+#  -c:v libx264 -pix_fmt yuv420p "${OUTPUT_DIR}/out.mp4" -y
+