File size: 531 Bytes
45d12c1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
# pip install diffusers==0.35.0 peft==0.17.0 transformers==4.56.0
# GPU 6 is faulty
export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5"
GPU_NUM=6 # 2,4,8
MODEL_PATH="./data/QwenImage"
OUTPUT_DIR="./data/qwenimage_rl_embeddings"
# Change to source_code directory if not already there
cd "$(dirname "$0")/../.."
torchrun --nproc_per_node=$GPU_NUM --master_port 19002 \
fastvideo/data_preprocess/preprocess_qwenimage_embedding.py \
--model_path $MODEL_PATH \
--output_dir $OUTPUT_DIR \
--prompt_dir "./assets/prompts.txt"
|