diff --git a/fooocus_version.py b/fooocus_version.py index 69337892..e76c5651 100644 --- a/fooocus_version.py +++ b/fooocus_version.py @@ -1 +1 @@ -version = '2.1.20' +version = '2.1.21' diff --git a/modules/expansion.py b/modules/expansion.py index acf9ddf7..86ec3e9f 100644 --- a/modules/expansion.py +++ b/modules/expansion.py @@ -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)