add nvidia/llama-3.2-nv-rerankqa-1b-v1 to set of known ranking models (…

…#1183)
deepset-ai · Nov 13, 2024 · a221ad5 · a221ad5
1 parent 051caad
commit a221ad5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/integrations/nvidia/src/haystack_integrations/components/rankers/nvidia/ranker.py b/integrations/nvidia/src/haystack_integrations/components/rankers/nvidia/ranker.py
@@ -12,6 +12,7 @@
 
 _MODEL_ENDPOINT_MAP = {
     "nvidia/nv-rerankqa-mistral-4b-v3": "https://ai.api.nvidia.com/v1/retrieval/nvidia/nv-rerankqa-mistral-4b-v3/reranking",
+    "nvidia/llama-3.2-nv-rerankqa-1b-v1": "https://ai.api.nvidia.com/v1/retrieval/nvidia/llama-3_2-nv-rerankqa-1b-v1/reranking",
 }