add v1 compat

michaelfeil · May 26, 2024 · ab6125d · ab6125d
1 parent f8a3682
commit ab6125d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/libs/infinity_emb/infinity_emb/infinity_server.py b/libs/infinity_emb/infinity_emb/infinity_server.py
@@ -340,10 +340,10 @@ def v1(
             batch_size=[batch_size],
             revision=[revision],  # type: ignore
             trust_remote_code=[trust_remote_code],
-            engine=engine,
-            dtype=dtype,
-            pooling_method=pooling_method,
-            device=device,
+            engine=[engine],
+            dtype=[dtype],
+            pooling_method=[pooling_method],
+            device=[device],
             model_warmup=[model_warmup],
             vector_disk_cache=[vector_disk_cache],
             lengths_via_tokenize=[lengths_via_tokenize],