Update cohere.py

stanfordnlp · Jun 12, 2024 · 86a7331 · 86a7331
1 parent a009cfd
commit 86a7331
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dsp/modules/cohere.py b/dsp/modules/cohere.py
@@ -63,7 +63,7 @@ def __init__(
         self.kwargs = {
             "model": model,
             "temperature": 0.0,
-            "max_tokens": 150,
+            "max_tokens": 2000,
             "p": 1,
             "num_generations": 1,
             **kwargs,