fix lowvram

This commit is contained in:
lvmin 2023-10-08 16:39:55 -07:00
parent a45e4d011e
commit 51b9bf5e7d
2 changed files with 5 additions and 1 deletions

View File

@ -1 +1 @@
version = '2.1.20'
version = '2.1.21'

View File

@ -50,6 +50,10 @@ class FooocusExpansion:
print(f'Fooocus Expansion engine loaded for {load_device}, use_fp16 = {use_fp16}.')
def __call__(self, prompt, seed):
if self.patcher.current_device != self.patcher.load_device:
print('Fooocus Expansion loaded by itself.')
model_management.load_model_gpu(self.patcher)
seed = int(seed)
set_seed(seed)
origin = safe_str(prompt)