diff --git a/vllm/executor/tpu_executor.py b/vllm/executor/tpu_executor.py index 54fc40c013261..0ac6f6ba91cf2 100644 --- a/vllm/executor/tpu_executor.py +++ b/vllm/executor/tpu_executor.py @@ -27,6 +27,8 @@ class TPUExecutor(ExecutorBase): self.parallel_config, self.scheduler_config, self.device_config, + self.cache_config, + self.vision_language_config, ) self.driver_worker.init_device() self.driver_worker.load_model()