Update model.py

This commit is contained in:
kijai
2025-11-06 21:01:03 +02:00
parent b1f8309a20
commit cf86f4f0a4
+1 -1
View File
@@ -2605,7 +2605,7 @@ class WanModel(torch.nn.Module):
if add_text_emb is not None:
self.text_projection.to(self.main_device)
add_text_emb = self.text_projection(add_text_emb.to(self.text_projection[0].weight.dtype)).to(text_embed_dtype)
context = torch.cat([context, add_text_emb], dim=1)
context = torch.cat([add_text_emb, context], dim=1)
context = self.text_embedding(context)
if self.is_longcat: