codelion commited on
Commit
c02d7fb
1 Parent(s): e1ed375

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -62,10 +62,10 @@ def generate(
62
  max_new_tokens=max_new_tokens,
63
  do_sample=True,
64
  top_p=top_p,
65
- top_k=top_k,
66
  temperature=temperature,
67
- num_beams=1,
68
- repetition_penalty=repetition_penalty,
69
  )
70
  t = Thread(target=model.generate, kwargs=generate_kwargs)
71
  t.start()
 
62
  max_new_tokens=max_new_tokens,
63
  do_sample=True,
64
  top_p=top_p,
65
+ #top_k=top_k,
66
  temperature=temperature,
67
+ #num_beams=1,
68
+ #repetition_penalty=repetition_penalty,
69
  )
70
  t = Thread(target=model.generate, kwargs=generate_kwargs)
71
  t.start()