Skip to content
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

Add NPU support for Llava #1446

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

HelloWorldBeginner
Copy link

[NPU] Support Llava training and inference for Ascend NPU.

I've modified some codes to add support for NPU, allowing LLAVA to perform both training and inference on NPU.
It works fine on NPU.
NPU Training
image
NPU Evaluate
image
NPU Inference
image

@HelloWorldBeginner
Copy link
Author

HelloWorldBeginner commented May 6, 2024

Hi @haotian-liu
Can you review my codes?

@MengqingCao
Copy link

cool! It helps for NPU users like me, thx!

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.

None yet

2 participants