From d854ed0bcf1284e76825912338a34871bf1bf029 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Thu, 3 Oct 2024 09:44:54 -0400 Subject: [PATCH] Allow using SD3 type te output on flux model. --- comfy/ldm/flux/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/ldm/flux/model.py b/comfy/ldm/flux/model.py index 5d20d888..b7d8a692 100644 --- a/comfy/ldm/flux/model.py +++ b/comfy/ldm/flux/model.py @@ -108,7 +108,7 @@ class Flux(nn.Module): raise ValueError("Didn't get guidance strength for guidance distilled model.") vec = vec + self.guidance_in(timestep_embedding(guidance, 256).to(img.dtype)) - vec = vec + self.vector_in(y) + vec = vec + self.vector_in(y[:,:self.params.vec_in_dim]) txt = self.txt_in(txt) ids = torch.cat((txt_ids, img_ids), dim=1)