跳到正文
原文
Hugging Face Blog·· 2023-05-15精选AI 评分60

RWKV 架构集成进 Hugging Face transformers 库

Introducing RWKV - An RNN with the advantages of a transformer

AI 导读

Hugging Face 官方博客宣布 RWKV 架构已通过 Pull Request 集成进 transformers 库,可通过源码安装或 main 分支使用,支持 pipeline 文本生成与 alpaca 风格的 RWKV-4 Raven 聊天模型调用,并提供 convert_rwkv_checkpoint_to_hf.py 脚本转换原始权重。

推荐理由

原文对比了 RWKV 与 Transformer、RNN 的取舍并给出可直接运行的代码,便于读者判断其适用场景。

来源:Hugging Face Blog · huggingface.co