Add support for RWKV in Hugging Face format

2023-05-23 02:07:28 -03:00 · 2023-05-23 02:07:28 -03:00 · cd3618d7fb
commit cd3618d7fb
parent 75adc110d4
4 changed files with 15 additions and 5 deletions
--- a/modules/shared.py
+++ b/modules/shared.py
@ -69,7 +69,7 @@ settings = {
        'default': 'Default',
        '.*(alpaca|llama|llava|vicuna)': "LLaMA-Precise",
        '.*pygmalion': 'NovelAI-Storywriter',
-        '.*RWKV': 'Naive',
+        '.*RWKV.*\.pth': 'Naive',
        '.*moss': 'MOSS',
    },
    'prompts': {