Spaces:
Running
Running
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 |
|