From a936472512e1cd9c45015c4f653bc7a6ed093a33 Mon Sep 17 00:00:00 2001 From: yin1991 <84140478+xiaokuili@users.noreply.github.com> Date: Sat, 27 Jan 2024 07:01:12 +0800 Subject: [PATCH] docs: Update documentation to use 'model_id' rather than 'model_name' to match actual API (#16615) - **Description:** Replace 'model_name' with 'model_id' for accuracy - **Issue:** [link-to-issue](https://github.com/langchain-ai/langchain/issues/16577) - **Dependencies:** - **Twitter handle:** --- .../embeddings/self_hosted_hugging_face.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/community/langchain_community/embeddings/self_hosted_hugging_face.py b/libs/community/langchain_community/embeddings/self_hosted_hugging_face.py index 0b706532cf..d45a802492 100644 --- a/libs/community/langchain_community/embeddings/self_hosted_hugging_face.py +++ b/libs/community/langchain_community/embeddings/self_hosted_hugging_face.py @@ -71,9 +71,9 @@ class SelfHostedHuggingFaceEmbeddings(SelfHostedEmbeddings): from langchain_community.embeddings import SelfHostedHuggingFaceEmbeddings import runhouse as rh - model_name = "sentence-transformers/all-mpnet-base-v2" + model_id = "sentence-transformers/all-mpnet-base-v2" gpu = rh.cluster(name="rh-a10x", instance_type="A100:1") - hf = SelfHostedHuggingFaceEmbeddings(model_name=model_name, hardware=gpu) + hf = SelfHostedHuggingFaceEmbeddings(model_id=model_id, hardware=gpu) """ client: Any #: :meta private: