From 8c12f34d4ee319bf5b379bbed67d01e6bded27c3 Mon Sep 17 00:00:00 2001
From: Aryan <aryan@huggingface.co>
Date: Thu, 17 Oct 2024 17:29:38 +0530
Subject: [PATCH] update (#42)

---
 prepare_dataset.sh | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/prepare_dataset.sh b/prepare_dataset.sh
index adacfa7..65def93 100755
--- a/prepare_dataset.sh
+++ b/prepare_dataset.sh
@@ -9,9 +9,10 @@ DATA_ROOT="/path/to/my/datasets/video-dataset"  # This needs to be the path to t
 CAPTION_COLUMN="prompt.txt"
 VIDEO_COLUMN="videos.txt"
 OUTPUT_DIR="/path/to/my/datasets/preprocessed-dataset"
-HEIGHT=480
-WIDTH=720
-MAX_NUM_FRAMES=49
+HEIGHT_BUCKETS="480 720"
+WIDTH_BUCKETS="720 960"
+FRAME_BUCKETS="49"
+MAX_NUM_FRAMES="49"
 MAX_SEQUENCE_LENGTH=226
 TARGET_FPS=8
 BATCH_SIZE=1
@@ -27,8 +28,9 @@ CMD_WITHOUT_PRE_ENCODING="\
       --caption_column $CAPTION_COLUMN \
       --video_column $VIDEO_COLUMN \
       --output_dir $OUTPUT_DIR \
-      --height $HEIGHT \
-      --width $WIDTH \
+      --height_buckets $HEIGHT_BUCKETS \
+      --width_buckets $WIDTH_BUCKETS \
+      --frame_buckets $FRAME_BUCKETS \
       --max_num_frames $MAX_NUM_FRAMES \
       --max_sequence_length $MAX_SEQUENCE_LENGTH \
       --target_fps $TARGET_FPS \