github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-19 12:12:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

modify shell for check

Browse Source
This commit is contained in:
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" # "torch_ddp" "torch_ddp_fp16"
for plugin in "low_level_zero" "gemini"; do 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" \ --pretrained_model_name_or_path="/data/dreambooth/diffuser/stable-diffusion-v1-4" \
--instance_data_dir="/data/dreambooth/Teyvat/data" \ --instance_data_dir="/data/dreambooth/Teyvat/data" \
--output_dir="./weight_output" \ --output_dir="./weight_output" \