Allow using SD3 type te output on flux model.

MoonRide303 · Oct 3, 2024 · d854ed0 · d854ed0
1 parent abcd006
commit d854ed0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/comfy/ldm/flux/model.py b/comfy/ldm/flux/model.py
@@ -108,7 +108,7 @@ def forward_orig(
                 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)