Whisper pipeline: cache models in python tests (#1389)

Ticket: 159277
openvinotoolkit · Dec 19, 2024 · e1f910d · e1f910d
1 parent 17f4eb3
commit e1f910d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/python_tests/test_whisper_generate_api.py b/tests/python_tests/test_whisper_generate_api.py
@@ -25,7 +25,9 @@ def run_gc_after_test():
     yield
     gc.collect()
 
-@functools.lru_cache(1)
+# used whisper models are relatively small
+# cache them in memory to speedup tests
+@functools.lru_cache(3)
 def read_whisper_model(params, **tokenizer_kwargs):
     model_id, path = params