Integrate Open Source Models from Hugging Face
Dify supports Text-Generation and Embeddings. Below are the corresponding Hugging Face model types:
Text-Generation:text-generation,text2text-generation
Embeddings:feature-extraction
The specific steps are as follows:
You need a Hugging Face account (registered address).
Set the API key of Hugging Face (obtain address).
Select a model to enter the Hugging Face model list page.
Dify supports accessing models on Hugging Face in two ways:
Hosted Inference API. This method uses the model officially deployed by Hugging Face. No fee is required. But the downside is that only a small number of models support this approach.
Inference Endpoint. This method uses resources such as AWS accessed by the Hugging Face to deploy the model and requires payment.
Models that access the Hosted Inference API
1 Select a model
Hosted inference API is supported only when there is an area containing Hosted inference API on the right side of the model details page. As shown in the figure below:
On the model details page, you can get the name of the model.
2 Using access models in Dify
Select Hosted Inference API for Endpoint Type in Settings > Model Provider > Hugging Face > Model Type
. As shown below:
API Token is the API Key set at the beginning of the article. The model name is the model name obtained in the previous step.
Method 2: Inference Endpoint
1 Select the model to deploy
Inference Endpoint is only supported for models with the Inference Endpoints option under the Deploy button on the right side of the model details page. As shown below:
2 Deployment model
Click the Deploy button for the model and select the Inference Endpoint option. If you have not bound a bank card before, you will need to bind the card. Just follow the process. After binding the card, the following interface will appear: modify the configuration according to the requirements, and click Create Endpoint in the lower left corner to create an Inference Endpoint.
After the model is deployed, you can see the Endpoint URL.
3 Using access models in Dify
Select Inference Endpoints for Endpoint Type in Settings > Model Provider > Hugging face > Model Type
. As shown below:
The API Token is the API Key set at the beginning of the article. The name of the Text-Generation model can be arbitrary, but the name of the Embeddings model needs to be consistent with Hugging Face.
The Endpoint URL is the Endpoint URL obtained after the successful deployment of the model in the previous step.
Note: The "User name / Organization Name" for Embeddings needs to be filled in according to your deployment method on Hugging Face's Inference Endpoints, with either the ''User name'' or the "Organization Name".
Last updated