Other model types

Besides text generation, text embedding, and text reranking, local Hugging Face connections can also serve other model types.

Image generation

The image generation capabilities are only available through the Dataiku Agent Management API. See the Developer Guide for tutorials using this feature.

Image-to-image mode is not available using local Hugging Face models.

Image embedding

Image embedding models are supported with local Hugging Face connections.

Image embedding can be used in Visual ML and through the Dataiku Agent Management API.

Other task-specific models

Other specialized local Hugging Face models can also be used in Dataiku Agent Management.

Examples include local classifiers used by Guardrails, such as prompt injection detection and toxicity detection models. See Prompt injection detection and Toxicity detection.