mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Remove RWKV loader (#5130)

Browse source
This commit is contained in:
oobabooga 2023-12-31 02:01:40 -03:00 • committed by GitHub
parent 0e54a09bcb
commit 2734ce3e4c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 3 additions and 189 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modules/models_settings.py
View file

@ -157,8 +157,6 @@ def infer_loader(model_name, model_settings):
loader = 'llama.cpp'
elif re.match(r'.*\.gguf', model_name.lower()):
loader = 'llama.cpp'
elif re.match(r'.*rwkv.*\.pth', model_name.lower()):
loader = 'RWKV'
elif re.match(r'.*exl2', model_name.lower()):
loader = 'ExLlamav2_HF'
elif re.match(r'.*-hqq', model_name.lower()):