#36 - Fix to_device() incorrectly casting integer tensors

Only apply dtype conversion to floating-point tensors, preserving
integer and boolean tensors for HuggingFace model compatibility.

Signed-off-by: Dpakkk <pokharelbikash981@gmail.com>

Author: Dpakkk

src/alpamayo_r1/helper.py

@@ -86,10 +86,10 @@ def to_device(
 ) -> Any:
     """Recursively cast data into the specified device, dtype."""
     if isinstance(data, torch.Tensor):
-        data = data.to(
-            device=device,
-            dtype=dtype,
-        )
+        if dtype is not None and data.is_floating_point():
+            data = data.to(device=device, dtype=dtype)
+        else:
+            data = data.to(device=device)
         return data
     elif isinstance(data, collections.abc.Mapping):
         return {key: to_device(data[key], device=device, dtype=dtype) for key in data}