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

Small change

Browse Source
This commit is contained in:
JerryZhou54
2025-06-10 23:17:06 +00:00
parent bd7b7078ba
commit 64c9c780df
3 changed files with 13 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/preprocess/v1_preprocess_wan_data_i2v.sh
+4 -3
View File
@@ -1,9 +1,10 @@
# export WANDB_MODE="offline"
export WANDB_MODE="offline"
export HOME="/mnt/weka/home/hao.zhang/wei"
GPU_NUM=1 # 2,4,8
MODEL_PATH="Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"
MODEL_TYPE="wan"
DATA_MERGE_PATH="finetrainers/crush-smol/merge.txt"
OUTPUT_DIR="crush-smol_preprocess"
DATA_MERGE_PATH="$HOME/FastVideo/data/wei-i2v-dataset/crush-smol_raw/merge.txt"
OUTPUT_DIR="$HOME/FastVideo/data/wei-i2v-dataset/crush-smol_preprocessed"
VALIDATION_PATH="assets/prompt.txt"
torchrun --nproc_per_node=$GPU_NUM \