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.
harrison/prompt_examples
Michael 2 years ago committed by GitHub
parent c7f9c62532
commit 6a3dca888b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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

Loading…
Cancel
Save