modify shell for check

1 year ago · 730a092ba2
parent 49567d56d1
commit 730a092ba2
1 changed files with 1 additions and 1 deletions
--- a/examples/images/dreambooth/colossalai.sh
+++ b/examples/images/dreambooth/colossalai.sh
@ -14,4 +14,4 @@ torchrun --nproc_per_node 4 --standalone train_dreambooth_colossalai.py \
  --lr_scheduler="constant" \
  --lr_warmup_steps=0 \
  --num_class_images=200 \
-  --placement="cuda" \
+  --placement="auto" \