From ae922dffe17f703b5a95b8815c9c30617bb3d938 Mon Sep 17 00:00:00 2001
From: Martin Forell <martin.forell@kit.edu>
Date: Tue, 8 Oct 2024 09:00:41 +0000
Subject: [PATCH] testing

---
 text-embedding-3-large.yaml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/text-embedding-3-large.yaml b/text-embedding-3-large.yaml
index 5818bc0..b3ff888 100644
--- a/text-embedding-3-large.yaml
+++ b/text-embedding-3-large.yaml
@@ -4,8 +4,9 @@ description: |
 config_file: |
     parameters:
       model: "nvidia/NV-Embed-v2"
-    backend: sentencetransformers
+    backend: transformers
     embeddings: true
+    type: "AutoModelForCausalLM"
     name: "text-embedding-3-large"
     trust_remote_code: true
 
-- 
GitLab