From 2113dac3442a2ff1416fb4c8c871c750a7073894 Mon Sep 17 00:00:00 2001 From: Matthew Farrellee Date: Tue, 22 Oct 2024 09:11:52 -0400 Subject: [PATCH] add ibm/granite-3.0-3b-a800m-instruct and ibm/granite-3.0-8b-instruct to set of supported chat models --- .../langchain_nvidia_ai_endpoints/_statics.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py index 5a6675a..e3f50c6 100644 --- a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py +++ b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py @@ -459,6 +459,16 @@ def validate_client(self) -> "Model": model_type="chat", client="ChatNVIDIA", ), + "ibm/granite-3.0-8b-instruct": Model( + id="ibm/granite-3.0-8b-instruct", + model_type="chat", + client="ChatNVIDIA", + ), + "ibm/granite-3.0-3b-a800m-instruct": Model( + id="ibm/granite-3.0-3b-a800m-instruct", + model_type="chat", + client="ChatNVIDIA", + ), } QA_MODEL_TABLE = {