diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-18 16:05:52 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-18 16:05:52 +0300 |
| commit | ec1924ee5789b72c31c65932b549c59ccae0cdd6 (patch) | |
| tree | e6a6357f8a54a91313cf53f8b6599e53e194f790 /modules/extras.py | |
| parent | e20b7e30fe17744acb74ad33c87c0963525ea921 (diff) | |
additional fix for difference model merging
Diffstat (limited to 'modules/extras.py')
| -rw-r--r-- | modules/extras.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/extras.py b/modules/extras.py index 03f6085e..b853fa5b 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()):
|