Skip to content

【开源实习】MindSpore自定义RWKV算子开发(Python接口实现) #1862

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 5 commits into from
Dec 23, 2024

Conversation

EliwiiKeeya
Copy link
Contributor

@EliwiiKeeya EliwiiKeeya commented Dec 11, 2024

任务链接: Gitee Issue IB4Z22

新增 RWKV6 Python接口的 mindspore 实现.

…framework.

This commit implements the definition of the RWKV6 model in the MindSpore framework, and also provides part of the definition of the model weight conversion and the model runtime script.
EliwiiKeeya and others added 4 commits December 20, 2024 11:26
Adjusted the file directory.
Use mindnlp.core.serialization.load interface to load weights.
Use mindnlp.core.nn instead of mindspore.nn.
Use mindnlp.core.ops instead of mindspore.ops.
Add download method to get vocab table for selection.
@lvyufeng lvyufeng merged commit f007978 into mindspore-lab:master Dec 23, 2024
4 of 5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants