Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-26 09:40:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Minor change

Browse Source
This commit is contained in:
oobabooga 2023-04-22 15:15:31 -03:00
parent 47666c4d00
commit c0b5c09860
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/models.py
View File

@ -40,7 +40,7 @@ if shared.args.deepspeed:
def find_model_type(model_name): def find_model_type(model_name):
model_name = model_name.lower() model_name = model_name.lower()
if 'rwkv-' in model_name.lower(): if 'rwkv-' in model_name:
return 'rwkv' return 'rwkv'
elif len(list(Path(f'{shared.args.model_dir}/{model_name}').glob('*ggml*.bin'))) > 0: elif len(list(Path(f'{shared.args.model_dir}/{model_name}').glob('*ggml*.bin'))) > 0:
return 'llamacpp' return 'llamacpp'