mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Enable NUMA feature for llama_cpp_python (#4040)

Browse source
This commit is contained in:
StoyanStAtanasov 2023-09-27 03:05:00 +02:00 • committed by GitHub
parent 87ea2d96fd
commit 7e6ff8d1f0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 21 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modules/loaders.py
View file

@ -100,6 +100,7 @@ loaders_and_params = OrderedDict({
'rope_freq_base',
'compress_pos_emb',
'cpu',
'numa',
],
'llamacpp_HF': [
'n_ctx',
@ -115,6 +116,7 @@ loaders_and_params = OrderedDict({
'rope_freq_base',
'compress_pos_emb',
'cpu',
'numa',
'cfg_cache',
'use_fast',
'llamacpp_HF_info',