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

@@ -106,9 +106,9 @@ class Flux(nn.Module):
if self.params.guidance_embed:
if guidance is None:
raise ValueError("Didn't get guidance strength for guidance distilled model.")
vec.add_(self.guidance_in(timestep_embedding(guidance, 256).to(img.dtype)))
vec = vec + self.guidance_in(timestep_embedding(guidance, 256).to(img.dtype))
vec.add_(self.vector_in(y))
vec = vec + self.vector_in(y)
txt = self.txt_in(txt)
ids = torch.cat((txt_ids, img_ids), dim=1)