github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-22 13:41:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

modify shell for check

Browse Source
This commit is contained in:
Maruyama_Aya 2023-06-07 17:23:01 +08:00
parent 4fc8bc68ac
commit c94a33579b
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,8 +6,9 @@ HF_DATASETS_OFFLINE=1
TRANSFORMERS_OFFLINE=1
DIFFUSERS_OFFLINE=1
for plugin in "torch_ddp" "torch_ddp_fp16" "gemini" "low_level_zero"; do
torchrun --nproc_per_node 4 --standalone train_dreambooth_colossalai.py \
# "torch_ddp" "torch_ddp_fp16"
for plugin in "low_level_zero" "gemini"; do
torchrun --nproc_per_node 8 --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" \