From 00be37e2b8f34345aec237ac0d4171992cf07c53 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Wed, 4 Oct 2023 14:53:10 -0700 Subject: [PATCH] use more recent way to write code (#541) --- modules/expansion.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/modules/expansion.py b/modules/expansion.py index e3658897..e5bf1b53 100644 --- a/modules/expansion.py +++ b/modules/expansion.py @@ -34,17 +34,14 @@ class FooocusExpansion: self.model.eval() load_device = model_management.text_encoder_device() + offload_device = model_management.text_encoder_offload_device() + use_fp16 = model_management.should_use_fp16(device=load_device) - if 'mps' in load_device.type: - load_device = torch.device('cpu') - - if 'cpu' not in load_device.type and model_management.should_use_fp16(): + if use_fp16: self.model.half() - offload_device = model_management.text_encoder_offload_device() self.patcher = ModelPatcher(self.model, load_device=load_device, offload_device=offload_device) - - print(f'Fooocus Expansion engine loaded for {load_device}.') + print(f'Fooocus Expansion engine loaded for {load_device}, use_fp16 = {use_fp16}.') def __call__(self, prompt, seed): seed = int(seed)