mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-04-13 04:27:02 +08:00
Minor
This commit is contained in:
parent
eb0a0466a9
commit
6692a30266
@ -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
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user