diff --git a/vllm_fl/platform.py b/vllm_fl/platform.py index cc2579fa..cab4175e 100644 --- a/vllm_fl/platform.py +++ b/vllm_fl/platform.py @@ -59,7 +59,7 @@ def is_cuda_alike(self) -> bool: def is_cuda(self) -> bool: """Stateless version of [torch.cuda.is_available][].""" - if self.vendor_name == "iluvatar": + if self.vendor_name == "iluvatar" or self.vendor_name == "hygon": return False return self.device_type == "cuda"