Add support for RWKV in Hugging Face format

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

View file

@ -37,7 +37,7 @@
"default": "Default",
".*(alpaca|llama|llava|vicuna)": "LLaMA-Precise",
".*pygmalion": "NovelAI-Storywriter",
".*RWKV": "Naive",
".*RWKV.*\.pth": "Naive",
".*moss": "MOSS"
},
"prompts": {