Using an embedding model from Hugging Face #2328
Replies: 2 comments
|
I think the issue is with the Hugging Face Inference Provider rather than the
I'd first verify whether the model has an available inference provider for the feature-extraction/embedding task in Hugging Face. If Also, I would keep the GraphRAG configuration conceptually as: embedding_models:
default_embedding_model:
model_provider: huggingface
model: BAAI/bge-small-en-v1.5
auth_method: api_key
api_key: ${HUGGINGFACE_API_KEY}rather than adding If you can share your GraphRAG and LiteLLM versions, I can check the exact Hugging Face provider configuration for those versions. |
|
I don't think the model identifier itself is the problem here. Your GraphRAG configuration is structurally correct: The error is coming from the Hugging Face inference layer rather than from GraphRAG's YAML parsing.
embedding_models:
default_embedding_model:
model_provider: huggingface
model: BAAI/bge-small-en-v1.5
auth_method: api_key
api_key: ${HUGGINGFACE_API_KEY}I would not add If the same error persists, I would test the model directly against Hugging Face first. That helps distinguish a GraphRAG/LiteLLM integration issue from an HF provider-availability issue. Another reliable option is to deploy the model as a dedicated Hugging Face Inference Endpoint and provide its URL through embedding_models:
default_embedding_model:
model_provider: huggingface
model: BAAI/bge-small-en-v1.5
auth_method: api_key
api_key: ${HUGGINGFACE_API_KEY}
api_base: https://<your-endpoint>.huggingface.cloudSo in short: I would keep the original provider/model split, update the GraphRAG/LiteLLM stack, and verify the HF inference route rather than changing the BGE model name. |
Uh oh!
There was an error while loading. Please reload this page.
I get the following error every time I attempt to use an embedding model from Hugging Face:
settings.yaml:I have tried:
huggingface/tomodelBAAI/frommodel/BAAItomodel_providerNothing seems to help. Has anyone faced this issue before? How did you resolve it?
All reactions