mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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
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'