HF Hub Chat Model Connector

This node establishes a connection to a specific chat model hosted on the Hugging Face Hub. The difference to the HF Hub LLM Connector is that this node allows you to provide prompt templates which are crucial for obtaining the best output from many models that have been fine-tuned for chatting. To use this node, you need to successfully authenticate with the Hugging Face Hub using the HF Hub Authenticator node.

Provide the name of the desired chat model repository available on the Hugging Face Hub as an input.

Options

Hugging Face Hub Settings

Repo ID

The model name to be used, in the format <organization_name>/<model_name>. For example, Writer/camel-5b-hf. You can find available models at the Hugging Face Models repository.

Model Task

Task type for a given model.

Please ensure that the model capabilities align with the chosen task.

Available options:

text-generation: A popular variant of Text Generation where the model predicts the next word given a sequence of words. GPT-based models, such as GPT-3, are commonly used for this task. Available text generation models
text2text-generation: Task used for mapping between pairs of texts, such as translation from one language to another. Available text-to-text generation models
summarization: Task used for generating text summaries. Available summarization models

Prompt Templates

System Prompt Template: Model specific system prompt template. Defaults to "%1". Refer to the Hugging Face Hub model card for information on the correct prompt template.
Prompt Template: Model specific prompt template. Defaults to "%1". Refer to the Hugging Face Hub model card for information on the correct prompt template.

Model Parameters

Top k

The number of top-k tokens to consider when generating text.

Typical p

The typical probability threshold for generating text.

Repetition penalty

The repetition penalty to use when generating text.

Max new tokens

The maximum number of tokens to generate in the completion.

The token count of your prompt plus max new tokens cannot exceed the model's context length.

Temperature

Sampling temperature to use, between 0.0 and 100.0. Higher values will make the output more random, while lower values will make it more focused and deterministic.

Top-p sampling

An alternative to sampling with temperature, where the model considers the results of the tokens (words) with top_p probability mass. Hence, 0.1 means only the tokens comprising the top 10% probability mass are considered.

Input Ports

: Validated authentication for Hugging Face Hub.

Output Ports

: Connection to a specific chat model from Hugging Face Hub.

Popular Predecessors

No recommendations found

Popular Successors

No recommendations found

Views

This node has no views

Workflows

No workflows found

Developers

You want to see the source code for this node? Click the following button and we’ll use our super-powers to find it for you.

Installation

To use this node in KNIME, install the extension KNIME AI Extension (Labs) from the below update site following our NodePit Product and Node Installation Guide:

v5.2

A zipped version of the software site can be downloaded here.

Plugin provider: KNIME AG, Zurich, Switzerland

Plugin version: 5.2.1.202402071416

On NodePit since: 2024-02-07

Last update: 2024-05-09

KNIME versions: Since v5.2

Deploy, schedule, execute, and monitor your KNIME workflows locally, in the cloud or on-premises – with our brand new NodePit Runner.

Try NodePit Runner!