From 6a3dca888ba75e219185c54cd0f0ba72d1f90a26 Mon Sep 17 00:00:00 2001 From: Michael Date: Thu, 27 Oct 2022 21:17:03 -0400 Subject: [PATCH] Fix cohere integration (#33) Currently the cohere module uses a non-supported model. Updating this to use the default model if one is not specified. --- langchain/llms/cohere.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/langchain/llms/cohere.py b/langchain/llms/cohere.py index cf6b9d9b989..5d642038198 100644 --- a/langchain/llms/cohere.py +++ b/langchain/llms/cohere.py @@ -22,7 +22,7 @@ class Cohere(BaseModel, LLM): """ client: Any #: :meta private: - model: str = "gptd-instruct-tft" + model: str = None """Model name to use.""" max_tokens: int = 256