Spaces:
Paused
Paused
harpreetsahota
commited on
Commit
•
9be8588
1
Parent(s):
f9a1fe1
reduced num_beams for beam search from 5 to 2
Browse files
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=
|
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
|