Merge pull request #588 from lllyasviel/dev

fix lowvram
This commit is contained in:
lllyasviel
2023-10-08 16:41:05 -07:00
committed by GitHub
2 changed files with 5 additions and 1 deletions
+1 -1
View File
@@ -1 +1 @@
version = '2.1.20' version = '2.1.21'
+4
View File
@@ -50,6 +50,10 @@ class FooocusExpansion:
print(f'Fooocus Expansion engine loaded for {load_device}, use_fp16 = {use_fp16}.') print(f'Fooocus Expansion engine loaded for {load_device}, use_fp16 = {use_fp16}.')
def __call__(self, prompt, seed): 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) seed = int(seed)
set_seed(seed) set_seed(seed)
origin = safe_str(prompt) origin = safe_str(prompt)