Comment on page
🤗 Hugging Face x Workbooks
Hugging Face models are supported in AI Workbooks!
Hugging Face (HF) is a open-source platform for sharing and utilizing ML models. It hosts thousands of fine-tuned machine learning and generative AI models. You can now use Hugging Face models in an AI Workbook - along with the other models we support!
Here are a few Hugging Face workbooks to get started:
- Click '+' to Add a cell
- Select 'Hugging Face Models'
- Select 'Task' (categorization of what the models do e.g. Text to Image)
This will create an HF cell with the default model for the selected Task. For 'Text to Image', the default model is stablilityai/stable-diffusion-2-1. See below:
You can now prompt the default HF model!
In order to update the model of the HF cell to a non-default model, you will need to enter your HF API token.
To add your HF API token, click the key icon next to the model name in the cell.
With your HF token added, you can update the model to any HF model for the selected Task which is hosted on HF Inference Endpoint!
You can check that a model will work for a cell by confirming that the Hosted inference API works in the Model Card in Hugging Face:
You cannot change a Hugging Face Cell to another model cell type (e.g. ChatGPT, Stable Diffusion) via the model dropdown menu. You will need to delete the HF Cell and create a new model cell.
We ♥️ Hugging Face. We believe that AI workbooks provide a great user experience to interact and play with Hugging Face models in a lightweight way.
Here are a few awesome things you can with Hugging Face models in Workbooks:
- 1.Try multiple Hugging Face models in a single notebook interface.
- 2.Organize, save, and share results from Hugging Face models.
- 3.Easily switch models to compare model performance for the same task