Add missing exllamav2 samplers

2023-11-16 07:09:40 -08:00 · 2023-11-16 07:09:40 -08:00 · 58c6001be9
commit 58c6001be9
parent cd41f8912b
2 changed files with 10 additions and 0 deletions
--- a/modules/loaders.py
+++ b/modules/loaders.py
@ -235,11 +235,16 @@ loaders_samplers = {
    'ExLlamav2': {
        'temperature',
        'top_p',
+        'min_p',
        'top_k',
        'typical_p',
+        'tfs',
        'repetition_penalty',
        'repetition_penalty_range',
        'seed',
+        'mirostat_mode',
+        'mirostat_tau',
+        'mirostat_eta',
        'ban_eos_token',
        'add_bos_token',
        'custom_token_bans',