Ganesh Karbhari commited on
Commit
b4be431
1 Parent(s): 8a01b2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -238,7 +238,7 @@ def format_prompt(message, history):
238
  return prompt
239
 
240
  def generate(
241
- prompt, history, temperature=0.3, max_new_tokens=3000, top_p=0.90,top_k=50
242
  ):
243
  temperature = float(temperature)
244
  if temperature < 1e-2:
@@ -249,7 +249,6 @@ def generate(
249
  temperature=temperature,
250
  max_new_tokens=max_new_tokens,
251
  top_p=top_p,
252
- top_k=top_k,
253
  #repetition_penalty=repetition_penalty,
254
  #do_sample=True,
255
  #seed=42,
 
238
  return prompt
239
 
240
  def generate(
241
+ prompt, history, temperature=0.3, max_new_tokens=3000, top_p=0.90
242
  ):
243
  temperature = float(temperature)
244
  if temperature < 1e-2:
 
249
  temperature=temperature,
250
  max_new_tokens=max_new_tokens,
251
  top_p=top_p,
 
252
  #repetition_penalty=repetition_penalty,
253
  #do_sample=True,
254
  #seed=42,