Spaces:

multimodalart
/

self-forcing

Running on Zero

multimodalart HF Staff commited on 1 day ago

Commit

6e7c357

verified ·

1 Parent(s): ec96039

Update wan/modules/causal_model.py

Files changed (1) hide show

wan/modules/causal_model.py CHANGED Viewed

@@ -236,6 +236,7 @@ class CausalWanSelfAttention(nn.Module):
         # output
         x = x.flatten(2)
         x = self.o(x)
         return x

         # output
         x = x.flatten(2)
+        x = x.to(self.o.weight.dtype)
         x = self.o(x)
         return x