Update Backend

Update Backend
This commit is contained in:
lllyasviel
2023-10-25 09:40:13 -07:00
parent bb965067e0
commit 38e70cebcc
11 changed files with 288 additions and 252 deletions
+1 -1
View File
@@ -156,7 +156,7 @@ class ControlNet(ControlBase):
context = cond['c_crossattn']
y = cond.get('c_adm', None)
y = cond.get('y', None)
if y is not None:
y = y.to(self.control_model.dtype)
control = self.control_model(x=x_noisy.to(self.control_model.dtype), hint=self.cond_hint, timesteps=t, context=context.to(self.control_model.dtype), y=y)