Embedding gpu (#14253)

2026-10-10 16:52:47 +03:00 · 2024-10-09 19:46:31 -06:00
parent 9fda259c0c
commit bc3a06178b
7 changed files with 34 additions and 33 deletions
@@ -118,7 +118,7 @@ class Embeddings:
            },
            embedding_function=jina_text_embedding_function,
            model_type="text",
-            preferred_providers=["CPUExecutionProvider"],
+            force_cpu=True,
        )

        self.vision_embedding = GenericONNXEmbedding(
@@ -130,7 +130,6 @@ class Embeddings:
            },
            embedding_function=jina_vision_embedding_function,
            model_type="vision",
-            preferred_providers=["CPUExecutionProvider"],
        )

    def _create_tables(self):