This commit is contained in:
Woosuk Kwon 2024-04-16 09:41:53 +00:00
parent eb0a0466a9
commit 6692a30266

View File

@ -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