diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index 0658f3f8fc..05c05ae9ff 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -556,10 +556,10 @@ from unsloth import DEVICE_TYPE
 def is_big_gpu(index) -> bool:
 
     if DEVICE_TYPE == "xpu":
-        prop = torch.xpu.get_device_properties(index)
+        prop = DeviceProperties.create(torch.device("xpu", index) if type(index) is int else index)
         min_sms = 16
     else:
-        prop = torch.cuda.get_device_properties(index)
+        prop = DeviceProperties.create(torch.device("cuda", index) if type(index) is int else index)
         min_sms = 80
 
     avail_sms = prop.multi_processor_count