🤗 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:

How To Use Hugging Face in AI Workbooks

Create a Hugging Face Model Cell

  • 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!

Update Cell to a Non-Default Hugging Face 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:

Update Hugging Face Cell to another Model Cell Type

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.

Not sure what Hugging Face models to use?

Check out the Hugging Face Models Library to explore models by task. Often times, model pages will include tips on prompts, examples, etc. so you can get a sense of the model.

If you want to request a task or model to be supported, please reach out to us on our Discord.

Benefits of Using Hugging Face x Workbooks

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

Feedback or Questions

We would love to hear from you! Please reach out to us on our Discord.

Last updated