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

added no_mmap & mlock parameters to llama.cpp and removed llamacpp_model_alternative (#1649)

Browse source 
---------

Co-authored-by: oobabooga <112222186+oobabooga@users.noreply.github.com>
This commit is contained in:
Ahmed Said 2023-05-03 00:25:28 +03:00 • committed by GitHub
parent 2f1a2846d1
commit fbcd32988e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 50 additions and 126 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modules/models.py
View file

@ -129,7 +129,7 @@ def load_model(model_name):
# llamacpp model
elif shared.model_type == 'llamacpp':
from modules.llamacpp_model_alternative import LlamaCppModel
from modules.llamacpp_model import LlamaCppModel
path = Path(f'{shared.args.model_dir}/{model_name}')
if path.is_file():