Nodes/ComfyUI Griptape Nodes/Griptape Embedding Driver: HuggingFace
ComfyUI Node

Griptape Embedding Driver: HuggingFace

Hugging Face Hub Embedding Driver

By griptape-ai·Created 2 years ago·Updated about a year ago· 238
Griptape Embedding Driver: HuggingFace
    • DRIVER
    embedding_modelsentence-transformers/all-MiniLM-L6-v2
    tokenizersentence-transformers/all-MiniLM-L6-v2
    max_output_tokens512
    api_token_env_varHUGGINGFACE_HUB_ACCESS_TOKEN
    CategoryGriptape/Agent Drivers/Embedding

    Inputs (4)

    NameTypeDefaultDescription
    embedding_modeloptSTRINGsentence-transformers/all-MiniLM-L6-v2The model to use for embeddings.
    tokenizeroptSTRINGsentence-transformers/all-MiniLM-L6-v2The tokenizer to use with the model.
    max_output_tokensoptINT512The maximum number of tokens for the output.
    api_token_env_varoptSTRINGHUGGINGFACE_HUB_ACCESS_TOKENEnvironment variable for the API token. Do not use your actual API key here.

    Outputs (1)

    NameTypeDescription
    DRIVEREMBEDDING_DRIVER