Spaces:
Running
Running
Commit
•
2fa02ef
1
Parent(s):
988a245
Update InferenceServer.py
Browse files- InferenceServer.py +3 -3
InferenceServer.py
CHANGED
|
@@ -47,9 +47,9 @@ def correct(input_sentence, max_candidates=1):
|
|
| 47 |
input_ids,
|
| 48 |
do_sample=True,
|
| 49 |
max_length=128,
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
num_beams=7,
|
| 53 |
early_stopping=True,
|
| 54 |
num_return_sequences=max_candidates)
|
| 55 |
|
|
|
|
| 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 |
|