Make flux controlnet work with sd3 text enc. (#8599)

This commit is contained in:
comfyanonymous 2025-06-19 15:50:05 -07:00 committed by GitHub
parent 91d40086db
commit 7e9267fa77
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -123,6 +123,8 @@ class ControlNetFlux(Flux):
if y is None:
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
img = self.img_in(img)