Add the top N-sigma sampler (#6796)

This commit is contained in:
oobabooga 2025-03-14 16:45:11 -03:00 committed by GitHub
parent 677d74a6a0
commit 5bcd2d7ad0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 60 additions and 3 deletions

View file

@ -302,6 +302,7 @@ def generate_reply_HF(question, original_question, seed, state, stopping_strings
'xtc_probability',
'tfs',
'top_a',
'top_n_sigma',
'dry_multiplier',
'dry_allowed_length',
'dry_base',