Skip to content

Commit

Permalink
docs: Update documentation to use 'model_id' rather than 'model_name'…
Browse files Browse the repository at this point in the history
… to match actual API (#16615)

- **Description:** Replace 'model_name' with 'model_id' for accuracy 
- **Issue:**
[link-to-issue](#16577)
  - **Dependencies:** 
  - **Twitter handle:**
  • Loading branch information
xiaokuili authored Jan 26, 2024
1 parent 6543e58 commit a936472
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit a936472

Please sign in to comment.