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

Support Mixtral quantization using INC #188

Closed
wants to merge 9 commits into from

Conversation

dudilester
Copy link

No description provided.

# Conflicts:
#	vllm/hpu/ops.py
#	vllm/model_executor/layers/quantization/inc.py
@Tiefen-boop Tiefen-boop force-pushed the dev/dlester/mixtral_main_1 branch 3 times, most recently from 53cdd9b to 5b515fa Compare August 29, 2024 09:45
@kzawora-intel kzawora-intel added the habana Issues or PRs submitted by Habana Labs label Aug 29, 2024
@Tiefen-boop Tiefen-boop force-pushed the dev/dlester/mixtral_main_1 branch 5 times, most recently from 0dc3495 to 4ea6b7c Compare September 2, 2024 12:56
vllm/hpu/ops.py Outdated Show resolved Hide resolved
vllm/worker/habana_model_runner.py Outdated Show resolved Hide resolved
@@ -13,9 +13,6 @@
from vllm.model_executor.utils import set_weight_attrs

Choose a reason for hiding this comment

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

@kzawora-intel could you check if these changes are upstreamable?

Copy link

@MrGeva MrGeva left a comment

Choose a reason for hiding this comment

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

LGTM :)

@@ -16,6 +16,7 @@

import habana_frameworks.torch as htorch
import torch
from neural_compressor.torch.quantization import finalize_calibration

Choose a reason for hiding this comment

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

This brakes taking fp8 measurements with TP>1 - please revert changes made to this file

@dudilester dudilester closed this Sep 11, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
habana Issues or PRs submitted by Habana Labs
Projects
None yet
Development

Successfully merging this pull request may close these issues.

7 participants