Spaces:
Running
Running
Commit
·
988a245
1
Parent(s):
0d586d2
Topp to Beam search
Browse files- InferenceServer.py +3 -2
InferenceServer.py
CHANGED
|
@@ -47,8 +47,9 @@ def correct(input_sentence, max_candidates=1):
|
|
| 47 |
input_ids,
|
| 48 |
do_sample=True,
|
| 49 |
max_length=128,
|
| 50 |
-
top_k=50,
|
| 51 |
-
top_p=0.95,
|
|
|
|
| 52 |
early_stopping=True,
|
| 53 |
num_return_sequences=max_candidates)
|
| 54 |
|
|
|
|
| 47 |
input_ids,
|
| 48 |
do_sample=True,
|
| 49 |
max_length=128,
|
| 50 |
+
# top_k=50,
|
| 51 |
+
# top_p=0.95,
|
| 52 |
+
num_beams=7,
|
| 53 |
early_stopping=True,
|
| 54 |
num_return_sequences=max_candidates)
|
| 55 |
|