[CI/Build] Fix multimodal tests (#22491)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
Cyrus Leung
2025-08-08 15:31:19 +08:00
committed by GitHub
parent 808a7b69df
commit 1712543df6
4 changed files with 17 additions and 15 deletions

View File

@@ -845,7 +845,8 @@ class LLMEngine:
def reset_mm_cache(self) -> bool:
"""Reset the multi-modal cache."""
return self.input_preprocessor.mm_registry.reset_processor_cache()
return self.input_preprocessor.mm_registry.reset_processor_cache(
self.model_config)
def reset_prefix_cache(self, device: Optional[Device] = None) -> bool:
"""Reset prefix cache for all devices."""