From 0f62757effb3ecc66f3f3a3d416c247f2d7b7c25 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 5 Oct 2023 11:30:32 -0700 Subject: [PATCH] fix (#547) --- fooocus_version.py | 2 +- modules/expansion.py | 6 ++++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/fooocus_version.py b/fooocus_version.py index c7df6d4c..79524f7a 100644 --- a/fooocus_version.py +++ b/fooocus_version.py @@ -1 +1 @@ -version = '2.0.91' +version = '2.0.92' diff --git a/modules/expansion.py b/modules/expansion.py index e5bf1b53..59a117ab 100644 --- 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()