Add custom sampler order support (#5443)

This commit is contained in:
oobabooga 2024-02-06 11:20:10 -03:00 committed by GitHub
parent 7301c7618f
commit 8c35fefb3b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 205 additions and 113 deletions

View file

@ -50,6 +50,7 @@ settings = {
'prompt_lookup_num_tokens': 0,
'custom_stopping_strings': '',
'custom_token_bans': '',
'sampler_priority': 'temperature,top_k,top_p,typical_p,epsilon_cutoff,eta_cutoff,tfs,top_a,min_p,dynamic_temperature,quadratic_sampling,mirostat',
'auto_max_new_tokens': False,
'ban_eos_token': False,
'add_bos_token': True,