diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-12-30 14:49:52 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-30 14:49:52 +0300 |
| commit | 908fb4ea71ef1cb8404cce705d749adec7615abc (patch) | |
| tree | d6f803154ab3a225722d2d751efa82af1fbbd91a /modules/sd_models_xl.py | |
| parent | c9c105c7dbdca65c0fe62f78cd9c9d41f9a5af1f (diff) | |
| parent | bfe418a58d39c69ca2672e7d8a1fd7ad2b34869b (diff) | |
Merge pull request #14390 from wangqyqq/sdxl-inpaint
Supporting for SDXL-Inpaint Model
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)
|