Add LADD (#45)

Co-authored-by: Peiyuan Zhang <a1286225768@gmail.com>
2024-11-15 19:28:41 -08:00
parent f8178a5a14
commit 897afb8b94
16 changed files with 1268 additions and 1060 deletions
@@ -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 \