Make flux controlnet work with sd3 text enc.

This commit is contained in:
comfyanonymous 2025-06-19 18:49:14 -04:00
parent 91d40086db
commit a9c028d969

View File

@ -123,6 +123,8 @@ class ControlNetFlux(Flux):
if y is None: if y is None:
y = torch.zeros((img.shape[0], self.params.vec_in_dim), device=img.device, dtype=img.dtype) y = torch.zeros((img.shape[0], self.params.vec_in_dim), device=img.device, dtype=img.dtype)
else:
y = y[:, :self.params.vec_in_dim]
# running on sequences img # running on sequences img
img = self.img_in(img) img = self.img_in(img)