Allow using SD3 type te output on flux model.

This commit is contained in:
comfyanonymous 2024-10-03 09:44:54 -04:00
parent abcd006b8c
commit d854ed0bcf
1 changed files with 1 additions and 1 deletions

View File

@ -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)