mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use more recent way to write code (#541)

Browse Source
This commit is contained in:
lllyasviel 2023-10-04 14:53:10 -07:00 committed by GitHub
parent 7fb0996461
commit 00be37e2b8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
modules/expansion.py
View File

@ -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)