nodes/banodoco-steerable-motion
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed control.py to work properly with ComfyUI updates

Browse Source
This commit is contained in:
Jedrzej Kosinski
2023-11-01 03:05:17 -05:00
parent 1b8800d783
commit 82fa785376
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
control/control.py
+4 -1
View File
@@ -235,7 +235,10 @@ class ControlNetAdvanced(ControlNet):
y = cond.get('c_adm', 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)
timestep = self.model_sampling_current.timestep(t)
x_noisy = self.model_sampling_current.calculate_input(t, x_noisy)
control = self.control_model(x=x_noisy.to(self.control_model.dtype), hint=self.cond_hint, timesteps=timestep.float(), context=context.to(self.control_model.dtype), y=y)
return self.control_merge(None, control, control_prev, output_dtype)
def apply_advanced_strengths_and_masks(self, x: Tensor, current_timestep_keyframe: TimestepKeyframe, batched_number: int):