add warmup to engine

michaelfeil · Jun 10, 2024 · 74f3722 · 74f3722
1 parent 0977109
commit 74f3722
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/infinity_emb/tests/unit_test/test_engine.py b/libs/infinity_emb/tests/unit_test/test_engine.py
@@ -157,7 +157,7 @@ async def test_torch_clip_embed():
         EngineArgs(
             model_name_or_path="wkcn/TinyCLIP-ViT-8M-16-Text-3M-YFCC15M",
             engine=InferenceEngine.torch,
-            model_warmup=False,
+            model_warmup=True,
         )
     )
     async with engine: