Fix dimension calculation issue in mask list during forward due to precision error (#12)

- Fix dimension calculation issue in mask list during forward due to precision error (6ebb839fb29560c7664f3e09431e7ce25ebc5b0c)


Co-authored-by: Z <[email protected]>

modeling_qwen2.py

@@ -130,7 +130,7 @@ class TimeSeriesEmbedding(nn.Module):
         batch_size = x.size(0)
         x = x.reshape(batch_size, -1, self.num_features)
 
-        mask = x[:, :, -1]
+        mask = x[:, :, -1].long()
         valid_lengths = mask.sum(dim=1).long()  # Shape: (batch_size)
 
         patch_cnt = (valid_lengths + self.patch_size - 1) // self.patch_size  # 向上取整