Skip to content

Commit dd344e0

Browse files
[Bugfix] Fix torch_xla in V0 which can't handle None seed introduced … (#14844)
Signed-off-by: Yarong Mu <ymu@google.com>
1 parent 54a8804 commit dd344e0

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

vllm/worker/tpu_worker.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,9 @@ def __init__(
5151
self.model_runner: TPUModelRunner = TPUModelRunner(
5252
vllm_config=vllm_config, is_driver_worker=is_driver_worker)
5353

54+
if self.model_config.seed is None:
55+
self.model_config.seed = 0
56+
5457
def init_device(self) -> None:
5558
os.environ["PJRT_DEVICE"] = "TPU"
5659
torch.set_grad_enabled(False)

0 commit comments

Comments
 (0)