Fix a bug in modules/shared.py

This commit is contained in:
oobabooga 2023-08-06 20:36:35 -07:00
parent 2cf64474f2
commit a373c96d59

View File

@ -199,6 +199,9 @@ if args.multi_user:
def fix_loader_name(name): def fix_loader_name(name):
if not name:
return name
name = name.lower() name = name.lower()
if name in ['llamacpp', 'llama.cpp', 'llama-cpp', 'llama cpp']: if name in ['llamacpp', 'llama.cpp', 'llama-cpp', 'llama cpp']:
return 'llama.cpp' return 'llama.cpp'