[LoRA] Support v1 LoRA training (#576)
This commit is contained in:
@@ -2,9 +2,9 @@
|
||||
GPU_NUM=1 # 2,4,8
|
||||
MODEL_PATH="Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
|
||||
MODEL_TYPE="wan"
|
||||
DATA_MERGE_PATH="data/crush-smol/merge.txt"
|
||||
OUTPUT_DIR="data/crush-smol/latents"
|
||||
VALIDATION_PATH="assets/prompt.txt"
|
||||
DATA_MERGE_PATH="data/mini_i2v_dataset/crush-smol_raw/merge.txt"
|
||||
OUTPUT_DIR="data/mini_i2v_dataset/crush-smol_preprocessed"
|
||||
VALIDATION_PATH="examples/training/finetune/wan_t2v_1_3b/crush_smol/validation.json"
|
||||
|
||||
torchrun --nproc_per_node=$GPU_NUM \
|
||||
fastvideo/v1/pipelines/preprocess/v1_preprocess.py \
|
||||
|
||||
Reference in New Issue
Block a user