Skip to content

support quant_lm_head arg in all WOQ configs #1881

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.

Sign up for GitHub

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

Merged
merged 15 commits into from
Jun 27, 2024
Merged

support quant_lm_head arg in all WOQ configs #1881

Changes from 1 commit
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
131f2cb
start from RTNConfig
xin3he Jun 19, 2024
b461291
fix bug
xin3he Jun 19, 2024
0ae3c7b
add UTs
xin3he Jun 19, 2024
39f649c
update UT
xin3he Jun 20, 2024
26b643d
enhance gptq forward as awq
xin3he Jun 20, 2024
b208ae3
update hqq
xin3he Jun 20, 2024
7073400
fix bug
xin3he Jun 20, 2024
98ae916
update doc and UTs for set_local
xin3he Jun 21, 2024
b8429a5
fix UTs
xin3he Jun 21, 2024
1104f44
fix bug
xin3he Jun 21, 2024
33bb948
add document for quant_lm_head
xin3he Jun 21, 2024
c7808e4
fix bug
xin3he Jun 21, 2024
ff99765
remove useless __all__
xin3he Jun 25, 2024
2351abb
update per review
xin3he Jun 25, 2024
39a6866
Merge branch 'master' into xinhe/quant_lm_head
xin3he Jun 25, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix bug 
Signed-off-by: xin3he <[email protected]>
  • Loading branch information
@xin3he
xin3he committed Jun 21, 2024
commit 1104f44aee13e5ca983890f19f0eba200023eefc
4 changes: 2 additions & 2 deletions test/3x/torch/quantization/weight_only/test_mixed_algos.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,8 @@ def run_fn(model):
class TestMixedTwoAlgo:
def test_mixed_gptq_and_rtn(self):
with patch.object(logger, "info") as mock_info:
rtn_config = RTNConfig(quant_lm_head=True)
gptq_config = GPTQConfig(double_quant_bits=4, white_list=["transformer.*"])
rtn_config = RTNConfig(white_list=[".*mlp.*"])
gptq_config = GPTQConfig(double_quant_bits=4, white_list=[".*attn.*"])
combined_config = rtn_config + gptq_config
logger.info(combined_config)

Expand Down