Update vllm/v1/worker/tpu_model_runner.py

Co-authored-by: Cyrus Leung <[email protected]>

Signed-off-by: NickLucche <[email protected]>

Author: NickLucche

vllm/v1/worker/tpu_model_runner.py

@@ -972,7 +972,7 @@ def initialize_kv_cache(self, kv_cache_config: KVCacheConfig) -> None:
 
     def reset_dynamo_cache(self):
         if self.is_multimodal_model:
-            compiled_model = self.model.get_language_model()
+            compiled_model = self.model.get_language_model().model
         else:
             compiled_model = self.model.model
         if isinstance(compiled_model, TorchCompileWrapperWithCustomDispatcher):