Spaces:
Runtime error
Runtime error
set default beam groups to 1
Browse files
app.py
CHANGED
|
@@ -128,8 +128,8 @@ def get_parser():
|
|
| 128 |
parser.add_argument(
|
| 129 |
"--num_beam_groups",
|
| 130 |
type=int,
|
| 131 |
-
default=
|
| 132 |
-
help="Number of groups to divide
|
| 133 |
)
|
| 134 |
return parser
|
| 135 |
|
|
|
|
| 128 |
parser.add_argument(
|
| 129 |
"--num_beam_groups",
|
| 130 |
type=int,
|
| 131 |
+
default=1,
|
| 132 |
+
help="Number of groups to divide best candidates into in order to ensure diversity among different groups of beams that yield the best n results. 1 means no group beam search. (default 1)",
|
| 133 |
)
|
| 134 |
return parser
|
| 135 |
|