storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-17 00:23:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

reuse base inference script

Browse Source
This commit is contained in:
Brandon Thomas
2025-01-19 22:22:16 -05:00
parent 7400f2876a
commit 5712d89e73
2 changed files with 404 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
animation/StableAnimator/command_basic_infer.sh
+4 -4
View File
@@ -5,16 +5,16 @@ OUTPUT_DIR="inference/output_1/final_output"
POSE_FRAME_DIR="inference/output_1/poses"
STARTING_IMAGE="inference/output_1/frames/frame_0.png"
mkdir -p OUTPUT_DIR
mkdir -p $OUTPUT_DIR
#CUDA_VISIBLE_DEVICES=0
python inference_basic.py \
--pretrained_model_name_or_path="checkpoints/SVD/stable-video-diffusion-img2vid-xt" \
--pretrained_model_name_or_path="checkpoints/stable-video-diffusion-img2vid-xt" \
--output_dir="${OUTPUT_DIR}" \
--validation_control_folder="${POSE_FRAME_DIR}" \
--validation_image="${STARTING_IMAGE}" \
--width=576 \
--height=1024 \
--width=1024 \
--height=576 \
--guidance_scale=3.0 \
--num_inference_steps=25 \
--posenet_model_name_or_path="checkpoints/Animation/pose_net.pth" \