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

updates.

Browse Source
This commit is contained in:
sayakpaul
2024-11-19 10:21:28 +05:30
parent ac83c78a5a
commit 1409d478d5
5 changed files with 22 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/mochi-1/prepare_dataset.sh
+3 -2
View File
@@ -11,11 +11,11 @@ VIDEO_COLUMN="videos.txt"
OUTPUT_DIR="/home/sayak/cogvideox-factory/video-dataset-disney/mochi-1/preprocessed-dataset"
HEIGHT_BUCKETS="480"
WIDTH_BUCKETS="848"
FRAME_BUCKETS="84"
FRAME_BUCKETS="1 84"
MAX_NUM_FRAMES="84"
MAX_SEQUENCE_LENGTH=256
TARGET_FPS=30
BATCH_SIZE=1
BATCH_SIZE=4
DTYPE=fp32
# To create a folder-style dataset structure without pre-encoding videos and captions
@@ -35,6 +35,7 @@ CMD_WITHOUT_PRE_ENCODING="\
--max_sequence_length $MAX_SEQUENCE_LENGTH \
--target_fps $TARGET_FPS \
--batch_size $BATCH_SIZE \
--use_slicing \
--dtype $DTYPE
"