add snowflake/arctic to set of ChatNVIDIA models

langchain-ai · Apr 24, 2024 · cf179e0 · cf179e0
1 parent 9609e21
commit cf179e0
Showing 1 changed file with 1 addition 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
@@ -139,6 +139,7 @@ class Model(BaseModel):
             "model_name": "meta/llama3-70b",
         },
         "ai-phi-3-mini": {"model_type": "chat", "model_name": "microsoft/phi-3-mini"},
+        "ai-arctic": {"model_type": "chat", "model_name": "snowflake/arctic"},
     }
 )