Skip to content

[Models] Llama 2 support#324

Merged
Aalanli merged 4 commits intohidet-org:mainfrom
Aalanli:llama2
Jul 22, 2023
Merged

[Models] Llama 2 support#324
Aalanli merged 4 commits intohidet-org:mainfrom
Aalanli:llama2

Conversation

@Aalanli
Copy link
Copy Markdown
Contributor

@Aalanli Aalanli commented Jul 21, 2023

Added some changes to support llama2. Not tested for numerical accuracy, waiting for reduce fix.

Allan Lin added 2 commits July 21, 2023 12:06
@yaoyaoding
Copy link
Copy Markdown
Member

Hi @Aalanli , could you check whether the numerical issue has been resolved with Xin's #325 ? Thanks.

@Aalanli
Copy link
Copy Markdown
Contributor Author

Aalanli commented Jul 22, 2023

Hi @yaoyaoding, yes, the numerical issues have been resolved. I noticed slight one character differences between torch and hidet output on llama2. But on a layer by layer basis, the activations are allclose.

@yaoyaoding
Copy link
Copy Markdown
Member

Thanks @Aalanli. It's fine that the output has very slight output difference. Feel free to merge this PR after passing the ci.

@Aalanli Aalanli merged commit 1b40ef4 into hidet-org:main Jul 22, 2023
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