Revert "Not sure if this actually changes anything but it can't hurt."
This reverts commit 34608de2e9.
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user