add ibm/granite-8b-code-instruct and ibm/granite-34b-code-instruct to…

… set of ChatNVIDIA models
langchain-ai · May 28, 2024 · 54b4a8a · 54b4a8a
1 parent f6b0662
commit 54b4a8a
Showing 1 changed file with 8 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
@@ -189,6 +189,14 @@ class Model(BaseModel):
             "model_type": "image_in",
             "model_name": "microsoft/phi-3-vision-128k-instruct",
         },
+        "ai-granite-8b-code-instruct": {
+            "model_type": "chat",
+            "model_name": "ibm/granite-8b-code-instruct",
+        },
+        "ai-granite-34b-code-instruct": {
+            "model_type": "chat",
+            "model_name": "ibm/granite-34b-code-instruct",
+        },
     }
 )