diff options
Diffstat (limited to 'model-integration/src/main/resources/configdefinitions/hugging-face-embedder.def')
-rw-r--r-- | model-integration/src/main/resources/configdefinitions/hugging-face-embedder.def | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/model-integration/src/main/resources/configdefinitions/hugging-face-embedder.def b/model-integration/src/main/resources/configdefinitions/hugging-face-embedder.def index 97515818f14..36957004e02 100644 --- a/model-integration/src/main/resources/configdefinitions/hugging-face-embedder.def +++ b/model-integration/src/main/resources/configdefinitions/hugging-face-embedder.def @@ -25,4 +25,5 @@ normalize bool default=false transformerExecutionMode enum { parallel, sequential } default=sequential transformerInterOpThreads int default=1 transformerIntraOpThreads int default=-4 -transformerGpuDevice int default=-1 +# GPU device id, -1 for CPU +transformerGpuDevice int default=0 |