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

Add gpu_split param to ExLlama

Browse source 
Adapted from code created by Ph0rk0z. Thank you Ph0rk0z.
This commit is contained in:
oobabooga 2023-06-16 20:49:36 -03:00
parent cb9be5db1c
commit 5f392122fd
6 changed files with 20 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

1
modules/loaders.py
View file

@ -53,6 +53,7 @@ loaders_and_params = {
'transformers_info'
],
'ExLlama' : [
'gpu_split',
'exllama_info',
]
}