From 5c94aaf290f8ad7bf4499a91c268ad0791b0432f Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Mon, 17 Oct 2022 08:28:18 +0300
Subject: fix bug for latest model merge RAM improvement

---
 modules/extras.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'modules')

diff --git a/modules/extras.py b/modules/extras.py
index 340a45fd..8dbab240 100644
--- a/modules/extras.py
+++ b/modules/extras.py
@@ -201,6 +201,7 @@ def run_modelmerger(primary_model_name, secondary_model_name, teritary_model_nam
         teritary_model = torch.load(teritary_model_info.filename, map_location='cpu')
         theta_2 = sd_models.get_state_dict_from_checkpoint(teritary_model)
     else:
+        teritary_model = None
         theta_2 = None
 
     theta_funcs = {
-- 
cgit v1.2.1