modify shell for check

2023-06-08 13:36:05 +08:00 · 2023-06-08 13:36:05 +08:00 · 49567d56d1
parent 039854b391
commit 49567d56d1
1 changed files with 1 additions and 1 deletions
--- a/examples/images/dreambooth/test_ci.sh
+++ b/examples/images/dreambooth/test_ci.sh
@ -20,5 +20,5 @@ for plugin in "gemini"; do
  --lr_scheduler="constant" \
  --lr_warmup_steps=0 \
  --num_class_images=200 \
-  --placement="cpu" # "cuda"
+  --placement="auto" # "cuda"
 done