dreambooth-altdiffusion

Configuration error

root commited on Dec 7, 2022

Commit

aadbd6e

•

1 Parent(s): f76cb2b

try mixed_precision

Files changed (1) hide show

train_dreambooth.py CHANGED Viewed

@@ -199,7 +199,7 @@ def parse_args():
     parser.add_argument(
         "--mixed_precision",
         type=str,
-        default="no",
         choices=["no", "fp16", "bf16"],
         help=(
             "Whether to use mixed precision. Choose"

     parser.add_argument(
         "--mixed_precision",
         type=str,
+        default="fp16",
         choices=["no", "fp16", "bf16"],
         help=(
             "Whether to use mixed precision. Choose"