File size: 643 Bytes
26601d1 3fc3d14 26601d1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
export CUDA_VISIBLE_DEVICES=6
EVAL_DIR=/home/lhchen/Sci-Fi/example_input_pairs
MODEL_NAME=CogVideoX-5b-I2V
OUT_DIR=outputs
mkdir -p $OUT_DIR
for example_dir in $(ls -d $EVAL_DIR/*)
do
example_name=$(EVAL_DIR $example_dir)
echo $example_name
out_fn=$OUT_DIR/$example_name'.mp4'
python Sci_Fi_frame_inbetweening.py \
--first_image=$example_dir/start.jpg \
--last_image=$example_dir/end.jpg \
--EF_Net_model_path='EF_Net/EF_Net.pt' \
--pretrained_model_name_or_path=$MODEL_NAME \
--prompt=$example_dir/prompt.txt \
--out_path=$out_fn \
--EF_Net_weights=1.0 \
done |