fix: OnnxRT EF now properly resolves ~ (#43)

Closes #35
amikos-tech · Sep 10, 2024 · fd5008b · fd5008b
1 parent 3a17034
commit fd5008b
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/chromadbx/embeddings/onnx.py b/chromadbx/embeddings/onnx.py
@@ -33,7 +33,8 @@ def __init__(
         ):
             raise ValueError("Preferred providers must be unique")
         self._preferred_providers = preferred_providers
-        self._model_path = model_path
+        self._model_path = os.path.expanduser(model_path)
+        print(self._model_path)
         self._max_length = max_length
         self._enabled_padding = enabled_padding
         try:

diff --git a/experiments/ort_ef.ipynb b/experiments/ort_ef.ipynb
@@ -27,7 +27,7 @@
     "from pathlib import Path\n",
     "from chromadbx.embeddings.onnx import OnnxRuntimeEmbeddings\n",
     "\n",
-    "ef=OnnxRuntimeEmbeddings(model_path=f\"{Path.home()}/.cache/chroma/onnx_models/all-MiniLM-L6-v2/onnx\",preferred_providers=[\"CPUExecutionProvider\"])\n",
+    "ef=OnnxRuntimeEmbeddings(model_path=f\"{Path.home()}/.cache/models/hf/all-MiniLM-L6-v2\",preferred_providers=[\"CPUExecutionProvider\"])\n",
     "\n",
     "ef([\"text\"])"
    ],
@@ -95,10 +95,24 @@
   {
    "metadata": {},
    "cell_type": "code",
-   "source": "",
+   "source": [
+    "from chromadbx.embeddings.onnx import OnnxRuntimeEmbeddings\n",
+    "\n",
+    "ef=OnnxRuntimeEmbeddings(model_path=f\"~/.cache/models/hf/all-MiniLM-L6-v2\",preferred_providers=[\"CPUExecutionProvider\"])\n",
+    "\n",
+    "ef([\"text\"])\n"
+   ],
    "id": "d0e2356d4fbfaac6",
    "outputs": [],
    "execution_count": null
+  },
+  {
+   "metadata": {},
+   "cell_type": "code",
+   "source": "",
+   "id": "498277298d41c6f3",
+   "outputs": [],
+   "execution_count": null
   }
  ],
  "metadata": {