From 51b9bf5e7de6eab384676c2e2fb55519c53a05d6 Mon Sep 17 00:00:00 2001
From: lvmin <lyuminzhang@outlook.com>
Date: Sun, 8 Oct 2023 16:39:55 -0700
Subject: [PATCH] fix lowvram

---
 fooocus_version.py   | 2 +-
 modules/expansion.py | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

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)