From cf179e01b43e98d76c9e8a5944299ee4aee44cf7 Mon Sep 17 00:00:00 2001 From: Matthew Farrellee Date: Wed, 24 Apr 2024 12:17:59 -0400 Subject: [PATCH] add snowflake/arctic to set of ChatNVIDIA models --- libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py | 1 + 1 file changed, 1 insertion(+) diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py index a81015e1..04400c4b 100644 --- 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"}, } )