Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-12-23 21:18:00 +01:00
Code Issues Packages Projects Releases 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
Show Stats 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
}
p = Path(model_name)
if p.exists():
model_name = p.parts[-1]
if loader is None:
if shared.args.loader is not None:
loader = shared.args.loader