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 | 4 |
1 files changed, 1 insertions, 3 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 index adc8f653168..584f23046ba 100644 --- 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 @@ -17,9 +17,6 @@ transformerAttentionMask string default=attention_mask # Output name transformerOutput string default=last_hidden_state -# GPU configuration -transformerGpuDevice int default=-1 -transformerGpuRequired bool default=false # Normalize tensors from tokenizer normalize bool default=false @@ -28,3 +25,4 @@ normalize bool default=false transformerExecutionMode enum { parallel, sequential } default=sequential transformerInterOpThreads int default=1 transformerIntraOpThreads int default=-4 +transformerGpuDevice int default=-1 |