ShieldX commited on
Commit
54cc37c
1 Parent(s): f500346

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,12 +29,12 @@ def predict(message, history):
29
  generation_config = GenerationConfig(
30
  penalty_alpha=0.6,
31
  early_stopping=True,
32
- num_beams=2,
33
  do_sample=True,
34
  top_k=5,
35
  temperature=0.5,
36
  repetition_penalty=1.2,
37
- # max_new_tokens=64,
38
  eos_token_id=tokenizer.eos_token_id,
39
  pad_token_id=tokenizer.eos_token_id
40
  )
 
29
  generation_config = GenerationConfig(
30
  penalty_alpha=0.6,
31
  early_stopping=True,
32
+ num_beams=4,
33
  do_sample=True,
34
  top_k=5,
35
  temperature=0.5,
36
  repetition_penalty=1.2,
37
+ max_new_tokens=128,
38
  eos_token_id=tokenizer.eos_token_id,
39
  pad_token_id=tokenizer.eos_token_id
40
  )