Allow using SD3 type te output on flux model.
This commit is contained in:
parent
abcd006b8c
commit
d854ed0bcf
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue