Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 18:45:09 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

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):
model_name = model_name.lower()
if 'rwkv-' in model_name.lower():
if 'rwkv-' in model_name:
return 'rwkv'
elif len(list(Path(f'{shared.args.model_dir}/{model_name}').glob('*ggml*.bin'))) > 0:
return 'llamacpp'