Downloads · 30 days
9
2% of all-time downloads
RWKV-4-Raven-3B-v11-zh
StarRing2022/RWKV-4-Raven-3B-v11-zh
RWKV-4-Raven-3B-v11-zh is a machine learning model from StarRing2022. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
将RWKV模型转化为HF格式，与HF无缝连接，几句代码调用RWKV 底座模型：RWKV-4-Raven-3B-v11-Eng49%-Chn49%-Jpn1%-Other1%-20230429-ctx4096.pth（https://huggingface.co/BlinkDL/rwkv-4-raven）
All-time downloads
421
Public
Repo size
11.9 GB
Likes
6
Public
Repo makeup
Click a slice to open those files.
.bin6 GB · 100%
At a glance
- Library
- transformers
- Model type
- rwkv
- Access
- Public
- Created
- May 23, 2023
- Updated
- Jul 17, 2023
- SHA
- 1e6e22e8
Model card
From the Hugging Face model README
将RWKV模型转化为HF格式，与HF无缝连接，几句代码调用RWKV 底座模型：RWKV-4-Raven-3B-v11-Eng49%-Chn49%-Jpn1%-Other1%-20230429-ctx4096.pth（https://huggingface.co/BlinkDL/rwkv-4-raven）
import torch
from transformers import GPTNeoXTokenizerFast, RwkvConfig, RwkvForCausalLM
model = RwkvForCausalLM.from_pretrained("StarRing2022/RWKV-4-Raven-3B-v11-zh")
tokenizer = GPTNeoXTokenizerFast.from_pretrained("StarRing2022/RWKV-4-Raven-3B-v11-zh")
text = "你好"
input_ids = tokenizer.encode(text, return_tensors='pt')
out = model.generate(input_ids=input_ids,max_new_tokens=128)
answer = tokenizer.decode(out[0])
print(answer)
GIT开源地址：https://github.com/StarRing2022/HF-For-RWKVRaven-Alpaca/
- Library
- transformers
- Type
- rwkv
- Created
- May 23, 2023
- Updated
- Jul 17, 2023