From 7e7e40f7e35da951546c5d8ab58bf0895cff4ec7 Mon Sep 17 00:00:00 2001 From: eesahe Date: Sat, 17 Aug 2024 19:04:59 +0000 Subject: [PATCH] Rollback minor change --- comfy/ldm/flux/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)