Update helper.py

helper.py

@@ -21,7 +21,6 @@ def get_fn(model_name: str, **model_kwargs) -> Callable:
         messages: list,  # Preprocessed messages from preprocess function
         temperature: float,
         max_tokens: int,
-        top_k: int,
         repetition_penalty: float,
         top_p: float
     ) -> Generator[str, None, None]:
@@ -32,7 +31,6 @@ def get_fn(model_name: str, **model_kwargs) -> Callable:
                 messages=messages,
                 temperature=temperature,
                 max_tokens=max_tokens,
-                top_k=top_k,
                 repetition_penalty=repetition_penalty,
                 top_p=top_p,
                 stream=True, 
@@ -131,7 +129,6 @@ def registry(name: str = None, **kwargs) -> gr.ChatInterface:
             messages=messages,
             temperature=temperature,
             max_tokens=max_tokens,
-            top_k=top_k,
             repetition_penalty=repetition_penalty,
             top_p=top_p
         )
@@ -153,7 +150,6 @@ def registry(name: str = None, **kwargs) -> gr.ChatInterface:
             ),
             gr.Slider(0.0, 1.0, value=0.7, label="Temperature"),
             gr.Slider(128, 4096, value=1024, label="Max new tokens"),
-            gr.Slider(1, 80, value=40, step=1, label="Top K sampling"),
             gr.Slider(0.0, 2.0, value=1.1, label="Repetition penalty"),
             gr.Slider(0.0, 1.0, value=0.95, label="Top P sampling"),
         ],