diff options
| author | brkirch <brkirch@users.noreply.github.com> | 2023-01-04 05:40:16 +0000 |
|---|---|---|
| committer | brkirch <brkirch@users.noreply.github.com> | 2023-01-06 05:14:20 +0000 |
| commit | f6ab5a39d762a7791573d1c52ae5a3024b10e8ed (patch) | |
| tree | c3958d77a6dae42457b571dbe0f1efec7ce45dd2 /modules/extras.py | |
| parent | d782a95967c9eea753df3333cd1954b6ec73eba0 (diff) | |
| parent | 3e22e294135ed0327ce9d9738655ff03c53df3c0 (diff) | |
| download | stable-diffusion-webui-gfx803-f6ab5a39d762a7791573d1c52ae5a3024b10e8ed.tar.gz stable-diffusion-webui-gfx803-f6ab5a39d762a7791573d1c52ae5a3024b10e8ed.tar.bz2 stable-diffusion-webui-gfx803-f6ab5a39d762a7791573d1c52ae5a3024b10e8ed.zip | |
Merge branch 'AUTOMATIC1111:master' into sub-quad_attn_opt
Diffstat (limited to 'modules/extras.py')
| -rw-r--r-- | modules/extras.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/extras.py b/modules/extras.py index 68939dea..5e270250 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -303,6 +303,8 @@ def run_modelmerger(primary_model_name, secondary_model_name, tertiary_model_nam theta_0[key][:, 0:4, :, :] = theta_func2(a[:, 0:4, :, :], b, multiplier)
result_is_inpainting_model = True
else:
+ assert a.shape == b.shape, f'Incompatible shapes for layer {key}: A is {a.shape}, and B is {b.shape}'
+
theta_0[key] = theta_func2(a, b, multiplier)
if save_as_half:
|