diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py index ab40673..5afdee9 100644 --- a/wanvideo/modules/model.py +++ b/wanvideo/modules/model.py @@ -153,7 +153,7 @@ def sinusoidal_embedding_1d(dim, position): # preprocess assert dim % 2 == 0 half = dim // 2 - position = position.type(torch.float64) + position = position.type(torch.float32) # calculation sinusoid = torch.outer(