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

Make --model work with argument like models/folder_name

Browse source
This commit is contained in:
oobabooga 2023-07-08 10:22:54 -07:00
parent acf24ebb49
commit 5ac4e4da8b
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
modules/models.py
View file

@ -61,6 +61,10 @@ def load_model(model_name, loader=None):
'ExLlama_HF': ExLlama_HF_loader 'ExLlama_HF': ExLlama_HF_loader
} }
p = Path(model_name)
if p.exists():
model_name = p.parts[-1]
if loader is None: if loader is None:
if shared.args.loader is not None: if shared.args.loader is not None:
loader = shared.args.loader loader = shared.args.loader