mirror of
https://github.com/storytold/FineTrainers-Conditioning.git
synced 2026-10-17 00:23:35 +00:00
sayakpaul
49 lines
1.5 KiB
Bash
49 lines
1.5 KiB
Bash
#!/bin/bash
|
|
|
|
MODEL_ID="genmo/mochi-1-preview"
|
|
|
|
NUM_GPUS=1
|
|
|
|
# For more details on the expected data format, please refer to the README.
|
|
DATA_ROOT="/home/sayak/cogvideox-factory/video-dataset-disney" # This needs to be the path to the base directory where your videos are located.
|
|
CAPTION_COLUMN="prompt.txt"
|
|
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"
|
|
MAX_NUM_FRAMES="84"
|
|
MAX_SEQUENCE_LENGTH=256
|
|
TARGET_FPS=30
|
|
BATCH_SIZE=1
|
|
DTYPE=fp32
|
|
|
|
# To create a folder-style dataset structure without pre-encoding videos and captions
|
|
# For Image-to-Video finetuning, make sure to pass `--save_image_latents`
|
|
CMD_WITHOUT_PRE_ENCODING="\
|
|
torchrun --nproc_per_node=$NUM_GPUS \
|
|
prepare_dataset.py \
|
|
--model_id $MODEL_ID \
|
|
--data_root $DATA_ROOT \
|
|
--caption_column $CAPTION_COLUMN \
|
|
--video_column $VIDEO_COLUMN \
|
|
--output_dir $OUTPUT_DIR \
|
|
--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 \
|
|
--batch_size $BATCH_SIZE \
|
|
--dtype $DTYPE
|
|
"
|
|
|
|
CMD_WITH_PRE_ENCODING="$CMD_WITHOUT_PRE_ENCODING --save_latents_and_embeddings"
|
|
|
|
# Select which you'd like to run
|
|
CMD=$CMD_WITH_PRE_ENCODING
|
|
|
|
echo "===== Running \`$CMD\` ====="
|
|
eval $CMD
|
|
echo -ne "===== Finished running script =====\n"
|