update

Author: jenniew

torchchat/cli/builder.py

@@ -68,10 +68,9 @@ class BuilderArgs:
 
     def __post_init__(self):
         if self.device is None:
-            # self.device = "cuda" if torch.cuda.is_available() else "cpu"
             if torch.cuda.is_available():
                 self.device = "cuda"
-            elif hasattr(torch, 'xpu') and torch.xpu.is_available():
+            elif torch.xpu.is_available():
                 self.device = "xpu"
             else:
                 self.device = "cpu"

torchchat/utils/device_info.py

@@ -38,13 +38,6 @@ def get_device_info(device: str) -> str:
     if device == "cuda":
         return torch.cuda.get_device_name(0)
     if device == "xpu":
-        # return (
-        #     check_output(
-        #         ["sycl-ls | grep gpu"], shell=True
-        #     )
-        #     .decode("utf-8")
-        #     .split("\n")[0]
-        #     )
         return (
             check_output(
                 ["xpu-smi discovery |grep 'Device Name:'"], shell=True