fix (#547)

2023-10-05 11:30:32 -07:00 · 2023-10-05 11:30:32 -07:00 · 0f62757eff
parent 6189e482bd
commit 0f62757eff
2 changed files with 7 additions and 1 deletions
--- a/fooocus_version.py
+++ b/fooocus_version.py
@ -1 +1 @@
-version = '2.0.91'
+version = '2.0.92'
--- a/modules/expansion.py
+++ b/modules/expansion.py
@ -37,6 +37,12 @@ class FooocusExpansion:
        offload_device = model_management.text_encoder_offload_device()
        use_fp16 = model_management.should_use_fp16(device=load_device)

+        # MPS hack
+        if model_management.is_device_mps(load_device):
+            load_device = torch.device('cpu')
+            offload_device = torch.device('cpu')
+            use_fp16 = False
+
        if use_fp16:
            self.model.half()