diff options
Diffstat (limited to 'model-integration/src/main/resources/configdefinitions/embedding.huggingface.hugging-face-embedder.def')
-rw-r--r-- | model-integration/src/main/resources/configdefinitions/embedding.huggingface.hugging-face-embedder.def | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/model-integration/src/main/resources/configdefinitions/embedding.huggingface.hugging-face-embedder.def b/model-integration/src/main/resources/configdefinitions/embedding.huggingface.hugging-face-embedder.def new file mode 100644 index 00000000000..cf7c3b336d6 --- /dev/null +++ b/model-integration/src/main/resources/configdefinitions/embedding.huggingface.hugging-face-embedder.def @@ -0,0 +1,18 @@ +namespace=embedding.huggingface + +# Path to tokenizer.json +tokenizerPath model + +# Path to model.onnx +transformerModel model + +# Max length of token sequence model can handle +transformerMaxTokens int default=512 + +# Input names +transformerInputIds string default=input_ids +transformerAttentionMask string default=attention_mask + +# Output name +transformerOutput string default=last_hidden_state + |