From ec1924ee5789b72c31c65932b549c59ccae0cdd6 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Tue, 18 Oct 2022 16:05:52 +0300
Subject: [PATCH] additional fix for difference model merging

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

diff --git a/modules/extras.py b/modules/extras.py
index 03f6085e4..b853fa5b2 100644
--- a/modules/extras.py
+++ b/modules/extras.py
@@ -220,7 +220,7 @@ def run_modelmerger(primary_model_name, secondary_model_name, teritary_model_nam
                     t2 = theta_2.get(key, torch.zeros_like(theta_1[key]))
                     theta_1[key] = theta_func1(theta_1[key], t2)
                 else:
-                    theta_1[key] = 0
+                    theta_1[key] = torch.zeros_like(theta_1[key])
     del theta_2, teritary_model
 
     for key in tqdm.tqdm(theta_0.keys()):