From 6692a30266fbcf670d1649b9117b4340c0138c0a Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Tue, 16 Apr 2024 09:41:53 +0000 Subject: [PATCH] Minor --- vllm/config.py | 1 - 1 file changed, 1 deletion(-) diff --git a/vllm/config.py b/vllm/config.py index 29e0f2dae11b3..7980b5e1aa544 100644 --- 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