nodes/hao-ai-lab-FastVideo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[feat]: Add LADD (#45)

Browse Source 
Co-authored-by: Peiyuan Zhang <a1286225768@gmail.com>
Co-authored-by: foreverpiano <pianoqwz@gmail.com>
This commit is contained in:
Zhang Peiyuan
2024-11-15 19:28:41 -08:00
co-authored by foreverpiano
parent 2f3a8661bf
commit 7106eadffc
16 changed files with 1268 additions and 1060 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/data_preprocess/finetune_data_gen.sh
+5 -1
View File
@@ -4,7 +4,7 @@ MODEL_PATH="/ephemeral/hao.zhang/outputfolder/ckptfolder/mochi_diffuser"
DATA_MERGE_PATH="/ephemeral/hao.zhang/resourcefolder/Mochi-Synthetic-Data-BW-Finetune/merge.txt"
OUTPUT_DIR="./data/BW-Finetune-Synthetic-Data_test"
torchrun --nproc_per_node=$GPU_NUM \
rchrun --nproc_per_node=$GPU_NUM \
./fastvideo/utils/data_preprocess/finetune_data_VAE.py \
--model_path $MODEL_PATH \
--data_merge_path $DATA_MERGE_PATH \
@@ -15,6 +15,10 @@ torchrun --nproc_per_node=$GPU_NUM \
--dataloader_num_workers 1 \
--output_dir=$OUTPUT_DIR
to
torchrun --nproc_per_node=$GPU_NUM \
./fastvideo/utils/data_preprocess/finetune_data_T5.py \
--model_path $MODEL_PATH \