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