Default to fp16 unet with 'auto'

2024-03-19 16:28:42 +02:00
parent 6a9ad93186
commit c6178d9fdc
1 changed files with 3 additions and 3 deletions
@@ -83,10 +83,10 @@ class DynamiCrafterModelLoader:
            self.model.eval()
            if dtype == "auto":
                try:
-                    if mm.should_use_bf16():
-                        self.model.to(convert_dtype('bf16'))
-                    elif mm.should_use_fp16():
+                    if mm.should_use_fp16():
                        self.model.to(convert_dtype('fp16'))
+                    elif mm.should_use_bf16():
+                        self.model.to(convert_dtype('bf16'))
                    else:
                        self.model.to(convert_dtype('fp32'))
                except: