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
commit 784161066e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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)