default values for kwargs

utils/generation.py

@@ -1,8 +1,8 @@
 from transformers import TextIteratorStreamer
 from threading import Thread
-from utils.tree_utils import get_docstrings, grab_before_comments
+from .tree_utils import get_docstrings, grab_before_comments
 
-def combine_generation_kwargs(temperature, max_new_tokens, top_p, repetition_penalty):
+def combine_generation_kwargs(temperature=2.0, max_new_tokens=512, top_p=0.95, repetition_penalty=1.2):
     """
     Combines the generation kwargs into a single dict.
     """