diff --git a/cohere/__init__.py b/cohere/__init__.py
index 744f63b7b..0ec1c7930 100644
--- a/cohere/__init__.py
+++ b/cohere/__init__.py
@@ -15,11 +15,13 @@ def __init__(self, api_key):
     self.api_url = COHERE_API_URL
     self.model = None
 
-  def generate(self, model, prompt, max_tokens=20, temperature=1):
+  def generate(self, model, prompt, max_tokens=20, temperature=1, k=0, p=0.75):
     json_body = json.dumps({
         "prompt": prompt,
         "max_tokens": max_tokens,
         "temperature": temperature,
+        "k": k,
+        "p": p,
       })
     response = self.__request(json_body, GENERATE_URL, model)
     return response["text"]