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

Add support for RWKV in Hugging Face format

Browse source
This commit is contained in:
oobabooga 2023-05-23 02:07:28 -03:00
parent 75adc110d4
commit cd3618d7fb
4 changed files with 15 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modules/models.py
View file

@ -45,7 +45,7 @@ def find_model_type(model_name):
return 'None'
model_name_lower = model_name.lower()
if 'rwkv-' in model_name_lower:
if re.match('.*rwkv.*\.pth', model_name_lower):
return 'rwkv'
elif len(list(path_to_model.glob('*ggml*.bin'))) > 0:
return 'llamacpp'