File size: 760 Bytes
fcb4edd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
#!bin/bash
noise_injection_steps=5
noise_injection_ratio=0.5
EVAL_DIR=examples
CHECKPOINT_DIR=checkpoints/svd_reverse_motion_with_attnflip
MODEL_NAME=stabilityai/stable-video-diffusion-img2vid-xt
OUT_DIR=results

mkdir -p $OUT_DIR
for example_dir in $(ls -d $EVAL_DIR/*)
do
    example_name=$(basename $example_dir)
    echo $example_name

    out_fn=$OUT_DIR/$example_name'.gif'
    python keyframe_interpolation.py \
        --frame1_path=$example_dir/frame1.png \
        --frame2_path=$example_dir/frame2.png \
        --pretrained_model_name_or_path=$MODEL_NAME \
        --checkpoint_dir=$CHECKPOINT_DIR \
        --noise_injection_steps=$noise_injection_steps \
        --noise_injection_ratio=$noise_injection_ratio \
        --out_path=$out_fn
done