codelion commited on
Commit
36fc847
1 Parent(s): 6d91895

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -64,8 +64,10 @@ def generate(
64
  top_p=top_p,
65
  #top_k=top_k,
66
  temperature=temperature,
 
 
67
  #num_beams=1,
68
- repetition_penalty=1.2,
69
  )
70
  t = Thread(target=model.generate, kwargs=generate_kwargs)
71
  t.start()
 
64
  top_p=top_p,
65
  #top_k=top_k,
66
  temperature=temperature,
67
+ eos_token_id=pipe.tokenizer.eos_token_id,
68
+ pad_token_id=pipe.tokenizer.pad_token_id,
69
  #num_beams=1,
70
+ #repetition_penalty=1.2,
71
  )
72
  t = Thread(target=model.generate, kwargs=generate_kwargs)
73
  t.start()