github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

modify shell for check

pull/3905/head
Maruyama_Aya 2023-06-08 11:15:10 +08:00
parent c94a33579b
commit cf4792c975
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/images/dreambooth/test_ci.sh
View File

@ -8,7 +8,7 @@ DIFFUSERS_OFFLINE=1
# "torch_ddp" "torch_ddp_fp16"
for plugin in "low_level_zero" "gemini"; do
torchrun --nproc_per_node 8 --standalone train_dreambooth_colossalai.py \
torchrun --nproc_per_node 4 --standalone train_dreambooth_colossalai.py \
--pretrained_model_name_or_path="/data/dreambooth/diffuser/stable-diffusion-v1-4" \
--instance_data_dir="/data/dreambooth/Teyvat/data" \
--output_dir="./weight_output" \