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

Add tensor split support for llama.cpp (#3171)

Browse source
This commit is contained in:
Shouyi 2023-07-26 07:59:26 +10:00 • committed by GitHub
parent f653546484
commit 031fe7225e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 20 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modules/loaders.py
View file

@ -33,6 +33,7 @@ loaders_and_params = {
'n_gqa',
'rms_norm_eps',
'n_gpu_layers',
'tensor_split',
'n_batch',
'threads',
'no_mmap',
@ -47,6 +48,7 @@ loaders_and_params = {
'n_gqa',
'rms_norm_eps',
'n_gpu_layers',
'tensor_split',
'n_batch',
'threads',
'no_mmap',