diff --git a/keybert/llm/_cohere.py b/keybert/llm/_cohere.py index fdf0d3d6..6a5dd5bf 100644 --- a/keybert/llm/_cohere.py +++ b/keybert/llm/_cohere.py @@ -82,7 +82,7 @@ class Cohere(BaseLLM): """ def __init__(self, client, - model: str = "xlarge", + model: str = "command", prompt: str = None, delay_in_seconds: float = None, verbose: bool = False