Minor

2024-04-16 09:41:53 +00:00
parent eb0a0466a9
commit 6692a30266
1 changed files with 0 additions and 1 deletions
--- a/vllm/config.py
+++ b/vllm/config.py
@ -636,7 +636,6 @@ class DeviceConfig:
        if self.device_type in ["neuron"]:
            self.device = torch.device("cpu")
        elif self.device_type in ["tpu"]:
-            # Will be set by `xm.xla_device()`
            self.device = None
        else:
            # Set device with device type