Skip to content

Commit c4ebd8d

Browse files
committed
fix: update greedy logic
1 parent fa3cc7c commit c4ebd8d

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

bigcodebench/generate.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -123,8 +123,7 @@ def main():
123123

124124
args = parser.parse_args()
125125

126-
if args.greedy and (args.temperature != 0 or args.bs != 1 or args.n_samples != 1)\
127-
or (args.temperature == 0 and args.n_samples == 1):
126+
if args.greedy or (args.temperature == 0 and args.n_samples == 1):
128127
args.temperature = 0
129128
args.bs = 1
130129
args.n_samples = 1

0 commit comments

Comments
 (0)