Make llama.cpp use tfs parameter

This commit is contained in:
oobabooga 2023-06-17 19:08:25 -03:00
parent e19cbea719
commit 05a743d6ad
2 changed files with 12 additions and 0 deletions

View File

@ -10,10 +10,21 @@ llama.cpp only uses the following parameters:
* top_p
* top_k
* repetition_penalty
* tfs
* mirostat_mode
* mirostat_tau
* mirostat_eta
### ExLlama
ExLlama only uses the following parameters:
* temperature
* top_p
* top_k
* repetition_penalty
* typical_p
### RWKV
RWKV only uses the following parameters when loaded through the old .pth weights:

View File

@ -68,6 +68,7 @@ class LlamaCppModel:
top_p=state['top_p'],
top_k=state['top_k'],
repeat_penalty=state['repetition_penalty'],
tfs_z=state['tfs'],
mirostat_mode=int(state['mirostat_mode']),
mirostat_tau=state['mirostat_tau'],
mirostat_eta=state['mirostat_eta'],