From 9feb034e343d6d7ef63395821658fb3774b30a24 Mon Sep 17 00:00:00 2001
From: wangqyqq <wangqyqq@163.com>
Date: Thu, 21 Dec 2023 20:15:51 +0800
Subject: support for sdxl-inpaint model

---
 modules/sd_models_xl.py | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'modules/sd_models_xl.py')

diff --git a/modules/sd_models_xl.py b/modules/sd_models_xl.py
index 01123321..d8a9a73b 100644
--- a/modules/sd_models_xl.py
+++ b/modules/sd_models_xl.py
@@ -34,6 +34,11 @@ 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.shape[1] == 9:
+        x = torch.cat([x] + cond['c_concat'], dim=1)
+
     return self.model(x, t, cond)
 
 
-- 
cgit v1.2.1