Revert "Not sure if this actually changes anything but it can't hurt."

This reverts commit 34608de2e9.
This commit is contained in:
comfyanonymous
2024-08-14 01:05:17 -04:00
parent 3e52e0364c
commit a5af64d3ce
3 changed files with 14 additions and 15 deletions

View File

@@ -64,8 +64,8 @@ class ControlNetFlux(Flux):
img = img + controlnet_cond
vec = self.time_in(timestep_embedding(timesteps, 256))
if self.params.guidance_embed:
vec.add_(self.guidance_in(timestep_embedding(guidance, 256)))
vec.add_(self.vector_in(y))
vec = vec + self.guidance_in(timestep_embedding(guidance, 256))
vec = vec + self.vector_in(y)
txt = self.txt_in(txt)
ids = torch.cat((txt_ids, img_ids), dim=1)