Skip to content

fix: stabilize HF export mask handling for e2e - #1384

Merged
fangyangci merged 3 commits into
release/v0.3.1from
fangyangci/e2e-fix-20260901
Sep 2, 2026
Merged

fix: stabilize HF export mask handling for e2e#1384
fangyangci merged 3 commits into
release/v0.3.1from
fangyangci/e2e-fix-20260901

Conversation

@fangyangci

@fangyangci fangyangci commented Sep 1, 2026

Copy link
Copy Markdown
Contributor

Summary

  • Resolve HF export/e2e regressions in QNN path.

What changed

  • Normalize binary attention_mask dummy inputs to bool in export IO.
  • Patch transformers compatibility handling for SDPA and mask helper signature drift.
  • Support kwargs invocation path in decoder wrapper.
  • Pass BLIP decoder attention mask as a 2D mask.

Validation

  • Focused unit regressions passed.
  • E2E passed: test_question_answering, test_fill_mask, test_image_to_text_fp16.

Closes #1385

@fangyangci
fangyangci requested a review from a team as a code owner September 1, 2026 16:10
@fangyangci
fangyangci changed the base branch from main to release/v0.3.1 September 1, 2026 16:15
…e2e-fix-20260901

# Conflicts:
#	src/winml/modelkit/models/hf/blip.py
#	tests/unit/export/test_blip_onnx_config.py
#	tests/unit/test_transformers_compat.py
Comment thread tests/unit/export/test_blip_onnx_config.py Fixed
Co-authored-by: Copilot Autofix powered by AI <62310815+github-advanced-security[bot]@users.noreply.github.com>
@fangyangci
fangyangci merged commit ffc9a36 into release/v0.3.1 Sep 2, 2026
9 checks passed
@fangyangci
fangyangci deleted the fangyangci/e2e-fix-20260901 branch September 2, 2026 02:46
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.

Fix e2e regressions in HF export mask handling for QNN path

2 participants