same workaround without --medvram cmd option case

wkpark · Jul 5, 2024 · 0e6a532 · 0e6a532
1 parent 32d5947
commit 0e6a532
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scripts/model_mixer.py b/scripts/model_mixer.py
@@ -4463,6 +4463,10 @@ def fake_checkpoint(checkpoint_info, metadata, model_name, sha256, fake=True):
             print("WARN: lowvram/medvram load_model() with minor workaround")
             sd_models.unload_model_weights()
             #sd_models.model_data.__init__()
+          elif shared.sd_model is not None:
+            # same workaround here.
+            print("WARN: workaround load_model() to fix 'copy out of meta tensor' error (launched without --medvram cmd option case.)")
+            sd_models.unload_model_weights()
 
           if shared.sd_model is not None:
             if partial_update: