Skip to content
This repository has been archived by the owner on Dec 1, 2024. It is now read-only.

Add support for Llama and Qwen models #135

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

marswen
Copy link

@marswen marswen commented Mar 29, 2024

This PR is to add support for Llama and Qwen models. Based on the scripts for OPT, RMSNorm and ROPE were added, and some parameters were adjusted for corresponding model architecture.

@lztjy
Copy link

lztjy commented Nov 18, 2024

我采纳了你的提交,在qwen测试的设置中默认把输入序列padding到128个tokens,当我修改这个输入序列长度时候会出现报错,是我没用对吗?

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants