Skip to content

fix: update cleanup logic for quantized model path in tests#1419

Merged
xin3he merged 1 commit intomainfrom
xinhe/2-9
Feb 9, 2026
Merged

fix: update cleanup logic for quantized model path in tests#1419
xin3he merged 1 commit intomainfrom
xinhe/2-9

Conversation

@xin3he
Copy link
Copy Markdown
Contributor

@xin3he xin3he commented Feb 9, 2026

Description

FAILED test_cuda/quantization/test_mxfp_nvfp.py::TestAutoRound::test_qwen_moe_quant_infer - RuntimeError: Error(s) in loading state_dict for Linear:
size mismatch for weight: copying a param with shape torch.Size([102400, 2048]) from checkpoint, the shape in current model is torch.Size([151936, 2048])

Type of Change

  • Bug fix
  • New feature
  • Documentation update
  • Performance improvement
  • Code refactoring
  • Other (please specify):

Related Issues

Fixes or relates to #1227

Checklist Before Submitting

  • My code has been tested locally.
  • Documentation has been updated as needed.
  • New or updated tests are included where applicable.

Signed-off-by: He, Xin3 <xin3.he@intel.com>
Copilot AI review requested due to automatic review settings February 9, 2026 05:18
@xin3he xin3he mentioned this pull request Feb 9, 2026
Copy link
Copy Markdown
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

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

Pull request overview

Updates test cleanup to remove the actual quantized model output directory rather than a hard-coded path, reducing test artifact leakage.

Changes:

  • Replace hard-coded ./saved cleanup with quantized_model_path cleanup in test_nvfp4_llmcompressor_format.
  • Add cleanup of quantized_model_path after quantized model saves in two additional NVFP4 MoE tests.

@chensuyue chensuyue added this to the 0.10.0 milestone Feb 9, 2026
@xin3he xin3he merged commit ebb534a into main Feb 9, 2026
15 checks passed
@xin3he xin3he deleted the xinhe/2-9 branch February 9, 2026 06:31
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.

4 participants