storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-22 08:05:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

update (#42)

Browse Source
This commit is contained in:
Aryan
2024-10-17 17:29:38 +05:30
committed by GitHub
parent feb2e26547
commit 8c12f34d4e
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
prepare_dataset.sh
+7 -5
View File
@@ -9,9 +9,10 @@ DATA_ROOT="/path/to/my/datasets/video-dataset" # This needs to be the path to t
CAPTION_COLUMN="prompt.txt" CAPTION_COLUMN="prompt.txt"
VIDEO_COLUMN="videos.txt" VIDEO_COLUMN="videos.txt"
OUTPUT_DIR="/path/to/my/datasets/preprocessed-dataset" OUTPUT_DIR="/path/to/my/datasets/preprocessed-dataset"
HEIGHT=480 HEIGHT_BUCKETS="480 720"
WIDTH=720 WIDTH_BUCKETS="720 960"
MAX_NUM_FRAMES=49 FRAME_BUCKETS="49"
MAX_NUM_FRAMES="49"
MAX_SEQUENCE_LENGTH=226 MAX_SEQUENCE_LENGTH=226
TARGET_FPS=8 TARGET_FPS=8
BATCH_SIZE=1 BATCH_SIZE=1
@@ -27,8 +28,9 @@ CMD_WITHOUT_PRE_ENCODING="\
--caption_column $CAPTION_COLUMN \ --caption_column $CAPTION_COLUMN \
--video_column $VIDEO_COLUMN \ --video_column $VIDEO_COLUMN \
--output_dir $OUTPUT_DIR \ --output_dir $OUTPUT_DIR \
--height $HEIGHT \ --height_buckets $HEIGHT_BUCKETS \
--width $WIDTH \ --width_buckets $WIDTH_BUCKETS \
--frame_buckets $FRAME_BUCKETS \
--max_num_frames $MAX_NUM_FRAMES \ --max_num_frames $MAX_NUM_FRAMES \
--max_sequence_length $MAX_SEQUENCE_LENGTH \ --max_sequence_length $MAX_SEQUENCE_LENGTH \
--target_fps $TARGET_FPS \ --target_fps $TARGET_FPS \