diff options
| author | bbc_mc <bbc_mc@hotmail.co.jp> | 2023-01-14 11:00:00 +0000 |
|---|---|---|
| committer | bbc_mc <bbc_mc@hotmail.co.jp> | 2023-01-14 11:09:32 +0000 |
| commit | 5f8685237ed6427c9a8e502124074c740ea7696a (patch) | |
| tree | c5eb90650522649317c5ce5c2381dd969848c2bf /modules/extras.py | |
| parent | 82725f0ac439f7e3b67858d55900e95330bbd326 (diff) | |
| download | stable-diffusion-webui-gfx803-5f8685237ed6427c9a8e502124074c740ea7696a.tar.gz stable-diffusion-webui-gfx803-5f8685237ed6427c9a8e502124074c740ea7696a.tar.bz2 stable-diffusion-webui-gfx803-5f8685237ed6427c9a8e502124074c740ea7696a.zip | |
Exclude clip index from merge
Diffstat (limited to 'modules/extras.py')
| -rw-r--r-- | modules/extras.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/modules/extras.py b/modules/extras.py index a03d558e..22668fcd 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -326,8 +326,14 @@ def run_modelmerger(primary_model_name, secondary_model_name, tertiary_model_nam print("Merging...")
+ chckpoint_dict_skip_on_merge = ["cond_stage_model.transformer.text_model.embeddings.position_ids"]
+
for key in tqdm.tqdm(theta_0.keys()):
if 'model' in key and key in theta_1:
+
+ if key in chckpoint_dict_skip_on_merge:
+ continue
+
a = theta_0[key]
b = theta_1[key]
@@ -352,6 +358,10 @@ def run_modelmerger(primary_model_name, secondary_model_name, tertiary_model_nam # I believe this part should be discarded, but I'll leave it for now until I am sure
for key in theta_1.keys():
if 'model' in key and key not in theta_0:
+
+ if key in chckpoint_dict_skip_on_merge:
+ continue
+
theta_0[key] = theta_1[key]
if save_as_half:
theta_0[key] = theta_0[key].half()
|