Commit 3783cb8b authored by Jedrzej Kosinski's avatar Jedrzej Kosinski
Browse files

change 'c_adm' to 'y' in ControlNet.get_control

parent d1d2fea8
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment