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 vllm awq loading logic #11950

Merged
merged 3 commits into from
Aug 28, 2024
Merged

Add vllm awq loading logic #11950

merged 3 commits into from
Aug 28, 2024

Conversation

gc-fu
Copy link
Contributor

@gc-fu gc-fu commented Aug 28, 2024

Description

Adding vLLM awq load logic.

We use an environment variable to get the group size. This is because within ipex-llm, we cannot get the quantization config.

1. Why the change?

2. User API changes

3. Summary of the change

4. How to test?

  • N/A
  • Unit test: Please manually trigger the PR Validation here by inputting the PR number (e.g., 1234). And paste your action link here once it has been successfully finished.
  • Application test
  • Document test
  • ...

5. New dependencies

  • New Python dependencies
    - Dependency1
    - Dependency2
    - ...
  • New Java/Scala dependencies and their license
    - Dependency1 and license1
    - Dependency2 and license2
    - ...

@gc-fu
Copy link
Contributor Author

gc-fu commented Aug 28, 2024

@gc-fu gc-fu requested a review from xiangyuT August 28, 2024 08:08
act_order=act_order,
enable_scale_search=enable_scale_search,
)
print("We are in ParallelLMHead branch")
Copy link
Contributor

Choose a reason for hiding this comment

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

Remove this print?

Copy link
Contributor

@xiangyuT xiangyuT left a comment

Choose a reason for hiding this comment

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

LGTM

@gc-fu gc-fu merged commit 0a7bd27 into main Aug 28, 2024
1 check passed
@liu-shaojun liu-shaojun deleted the add-awq-loading branch September 4, 2024 07:42
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