change cpu mode default to float

YunCheng66 · Mar 20, 2023 · 1885984 · 1885984
1 parent 8ca2ea6
commit 1885984
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/model.py b/modules/model.py
@@ -15,7 +15,7 @@ def prepare_model():
         elif cmd_opts.precision == "bf16":
             model = model.bfloat16()
         else:
-            model = model.bfloat16()
+            model = model.float()
     else:
         if cmd_opts.precision == "fp16":
             model = model.half().cuda()