Add --mul_mat_q param

This commit is contained in:
oobabooga 2023-08-18 07:58:20 -07:00
parent 28cf5862af
commit 4ec42679e3
6 changed files with 6 additions and 0 deletions

View file

@ -69,6 +69,7 @@ class LlamaCppModel:
'n_batch': shared.args.n_batch,
'use_mmap': not shared.args.no_mmap,
'use_mlock': shared.args.mlock,
'mul_mat_q': shared.args.mul_mat_q,
'low_vram': shared.args.low_vram,
'n_gpu_layers': shared.args.n_gpu_layers,
'rope_freq_base': 10000 * shared.args.alpha_value ** (64 / 63.),