Skip to content

Commit 66e16a0

Browse files
[Bugfix] Fix torch_xla which can't handle None seed introduced in #14274 (#14459)
Signed-off-by: Yarong Mu <ymu@google.com>
1 parent e1f0835 commit 66e16a0

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

vllm/v1/worker/tpu_worker.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,9 @@ def __init__(
7575
self.profile_dir)
7676
self.profiler = xp.start_server(9012)
7777

78+
if self.model_config.seed is None:
79+
self.model_config.seed = 0
80+
7881
def init_device(self):
7982
os.environ["PJRT_DEVICE"] = "TPU"
8083
torch.set_grad_enabled(False)

0 commit comments

Comments
 (0)