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 DSS 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 DSS API.

Other task-specific models

Other specialized local Hugging Face models can also be used in DSS.

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