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

updates

Browse Source
This commit is contained in:
sayakpaul
2024-11-29 10:48:54 +05:30
parent 2fde026d30
commit ced8558eb0
7 changed files with 347 additions and 698 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/mochi-1/prepare_dataset.sh
+4 -2
View File
@@ -1,9 +1,11 @@
#!/bin/bash
GPU_ID=0
VIDEO_DIR=/home/sayak/cogvideox-factory/video-dataset-disney-organized
VIDEO_DIR=video-dataset-disney-organized
OUTPUT_DIR=videos_prepared
NUM_FRAMES=37
RESOLUTION=480x848
python trim_and_crop_videos.py $VIDEO_DIR $OUTPUT_DIR --num_frames=37 --resolution=480x848 --force_upsample
python trim_and_crop_videos.py $VIDEO_DIR $OUTPUT_DIR --num_frames=$NUM_FRAMES --resolution=$RESOLUTION --force_upsample
CUDA_VISIBLE_DEVICES=$GPU_ID python embed.py $OUTPUT_DIR --shape=37x480x848