Skip to content

Commit 3c310d3

Browse files
Add min_p to generate args
1 parent b6f8683 commit 3c310d3

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

mamba_ssm/utils/generation.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -251,6 +251,7 @@ def generate(
251251
max_length,
252252
top_k=1,
253253
top_p=0.0,
254+
min_p=0.0,
254255
temperature=1.0,
255256
return_dict_in_generate=False,
256257
output_scores=False,

0 commit comments

Comments
 (0)