From 5df10b5624272ae6e84e3340a847543a1af6d38d Mon Sep 17 00:00:00 2001 From: fenyuan-gg Date: Thu, 30 Oct 2025 16:18:33 -0700 Subject: [PATCH] Update tpu_worker_jax.py update the location of init_cached_hf_modules Signed-off-by: fenyuan-gg --- tpu_inference/worker/tpu_worker_jax.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tpu_inference/worker/tpu_worker_jax.py b/tpu_inference/worker/tpu_worker_jax.py index 746999f3b..86e1d4555 100644 --- a/tpu_inference/worker/tpu_worker_jax.py +++ b/tpu_inference/worker/tpu_worker_jax.py @@ -87,7 +87,7 @@ def __init__(self, if self.model_config.trust_remote_code: # note: lazy import to avoid importing torch before initializing - from vllm.utils import init_cached_hf_modules + from vllm.utils.import_utils import init_cached_hf_modules init_cached_hf_modules()