diff --git a/comfy/ldm/flux/model.py b/comfy/ldm/flux/model.py index 1a75b6942..810d91f4d 100644 --- a/comfy/ldm/flux/model.py +++ b/comfy/ldm/flux/model.py @@ -133,9 +133,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)