Removed meta tensor usage

Author: MoonRide303

fooocus_version.py

@@ -1,2 +1,2 @@
-version = '2.0.78.4 MRE'
+version = '2.0.78.5 MRE'
 full_version = 'Fooocus ' + version

modules/core.py

@@ -56,14 +56,6 @@ def __init__(self, unet, vae, clip, clip_vision, model_filename=None):
         self.clip = clip
         self.clip_vision = clip_vision
 
-    def to_meta(self):
-        if self.unet is not None:
-            self.unet.model.to('meta')
-        if self.clip is not None:
-            self.clip.cond_stage_model.to('meta')
-        if self.vae is not None:
-            self.vae.first_stage_model.to('meta')
-
 
 @torch.no_grad()
 @torch.inference_mode()

modules/default_pipeline.py

@@ -44,9 +44,11 @@ def refresh_base_model(name):
         return
 
     if xl_base is not None:
-        xl_base.to_meta()
         xl_base = None
 
+    if xl_base_patched is not None:
+        xl_base_patched = None
+
     xl_base = core.load_model(filename)
     if not isinstance(xl_base.unet.model, BaseModel):
         print(f'Model not supported: {name}, using default base model instead.')
@@ -91,7 +93,6 @@ def refresh_refiner_model(name):
         return
 
     if xl_refiner is not None:
-        xl_refiner.to_meta()
         xl_refiner = None
 
     xl_refiner = core.load_model(filename)
@@ -105,7 +106,6 @@ def refresh_refiner_model(name):
     xl_refiner_hash = model_hash
     print(f'Refiner model loaded: {model_hash}')
 
-    xl_refiner.vae.first_stage_model.to('meta')
     xl_refiner.vae = None
     return
 

update_log_mre.md

@@ -1,3 +1,7 @@
+### 2.0.78.5 MRE
+
+* Removed meta tensor usage.
+
 ### 2.0.78.4 MRE
 
 * Added Load Last Prompt button (contribution from sngazm).