diff options
| author | kurisu_u <73207840+lanyeeee@users.noreply.github.com> | 2023-12-30 13:47:59 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-30 13:47:59 +0000 |
| commit | d05f9e8124160d5ef6fcea6523585f613f375fac (patch) | |
| tree | c2898250b3f96ccf791c9b4ebf7faf7f12f1a128 /modules/sd_models_xl.py | |
| parent | c069c2c5628728c9506dd034ef98e6335fd5bb34 (diff) | |
| parent | adcd65ba3493fc91c9d7c843d7e14275ad6fd881 (diff) | |
| download | stable-diffusion-webui-gfx803-d05f9e8124160d5ef6fcea6523585f613f375fac.tar.gz stable-diffusion-webui-gfx803-d05f9e8124160d5ef6fcea6523585f613f375fac.tar.bz2 stable-diffusion-webui-gfx803-d05f9e8124160d5ef6fcea6523585f613f375fac.zip | |
Merge branch 'dev' into api_thread_safe
Diffstat (limited to 'modules/sd_models_xl.py')
| -rw-r--r-- | modules/sd_models_xl.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/modules/sd_models_xl.py b/modules/sd_models_xl.py index 11259a36..1de31b0d 100644 --- a/modules/sd_models_xl.py +++ b/modules/sd_models_xl.py @@ -34,6 +34,12 @@ def get_learned_conditioning(self: sgm.models.diffusion.DiffusionEngine, batch: def apply_model(self: sgm.models.diffusion.DiffusionEngine, x, t, cond):
+ sd = self.model.state_dict()
+ diffusion_model_input = sd.get('diffusion_model.input_blocks.0.0.weight', None)
+ if diffusion_model_input is not None:
+ if diffusion_model_input.shape[1] == 9:
+ x = torch.cat([x] + cond['c_concat'], dim=1)
+
return self.model(x, t, cond)
|