Doc fix

intel-staging · May 22, 2024 · 311217c · 311217c
1 parent 9d870a5
commit 311217c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/docs/integrations/text_embedding/ipex_llm.ipynb b/docs/docs/integrations/text_embedding/ipex_llm.ipynb
@@ -90,7 +90,7 @@
     "print(f\"text_embeddings[0][:10]: {text_embeddings[0][:10]}\")\n",
     "print(f\"text_embeddings[1][:10]: {text_embeddings[1][:10]}\")\n",
     "\n",
-    "query_embedding = embedding_model.query(query)\n",
+    "query_embedding = embedding_model.embed_query(query)\n",
     "print(f\"query_embedding[:10]: {query_embedding[:10]}\")"
    ]
   }

diff --git a/docs/docs/integrations/text_embedding/ipex_llm_gpu.ipynb b/docs/docs/integrations/text_embedding/ipex_llm_gpu.ipynb
@@ -153,7 +153,7 @@
     "print(f\"text_embeddings[0][:10]: {text_embeddings[0][:10]}\")\n",
     "print(f\"text_embeddings[1][:10]: {text_embeddings[1][:10]}\")\n",
     "\n",
-    "query_embedding = embedding_model.query(query)\n",
+    "query_embedding = embedding_model.embed_query(query)\n",
     "print(f\"query_embedding[:10]: {query_embedding[:10]}\")"
    ]
   }