add zyphra/zamba2-7b-instruct to set of supported chat models

langchain-ai · Oct 15, 2024 · 7f3a1cc · 7f3a1cc
1 parent f0db608
commit 7f3a1cc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
@@ -454,6 +454,11 @@ def validate_client(self) -> "Model":
         client="ChatNVIDIA",
         supports_structured_output=True,
     ),
+    "zyphra/zamba2-7b-instruct": Model(
+        id="zyphra/zamba2-7b-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+    ),
 }
 
 QA_MODEL_TABLE = {