harpreetsahota commited on
Commit
9be8588
1 Parent(s): f9a1fe1

reduced num_beams for beam search from 5 to 2

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def generate_model_response(message: str) -> str:
52
  # Include **generate_kwargs to include the user-defined options
53
  output = model.generate(**inputs,
54
  max_new_tokens=3000,
55
- num_beams=5,
56
  no_repeat_ngram_size=4,
57
  early_stopping=True,
58
  do_sample=True
 
52
  # Include **generate_kwargs to include the user-defined options
53
  output = model.generate(**inputs,
54
  max_new_tokens=3000,
55
+ num_beams=2,
56
  no_repeat_ngram_size=4,
57
  early_stopping=True,
58
  do_sample=True