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 MoE and Compressed Inference Examples #160

Merged
merged 9 commits into from
Sep 11, 2024
Merged

Add MoE and Compressed Inference Examples #160

merged 9 commits into from
Sep 11, 2024

Conversation

Satrat
Copy link
Contributor

@Satrat Satrat commented Sep 10, 2024

SUMMARY:

  • Adding an additional MoE example for deepseek
  • Adding an example for compressed inference using an fp8 model

TEST PLAN:
Confirmed scripts run locally

@Satrat Satrat requested review from dsikka and rahul-tuli September 10, 2024 21:00
Copy link

👋 Hi! Thank you for contributing to llm-compressor. Please add the ready label when the PR is ready for review.

Copy link
Collaborator

@dsikka dsikka left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We'll want to automate these examples like the rest of them and Domenic has been working on that here: https://github.com/vllm-project/llm-compressor/pull/149/files

Could you add a follow-up ticket and add it to the backlog on adding these two cases to the automated examples as well? You can assign them to me

@dsikka
Copy link
Collaborator

dsikka commented Sep 10, 2024

Actually, scratch that, we should pull these in now

@Satrat Satrat merged commit 5bfb497 into main Sep 11, 2024
6 of 7 checks passed
@Satrat Satrat deleted the sa/moe_example branch September 11, 2024 20:34
markmc pushed a commit to markmc/llm-compressor that referenced this pull request Nov 13, 2024
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