Skip to content

Remove references to torchao's AffineQuantizedTensor#45321

Open
andrewor14 wants to merge 2 commits intohuggingface:mainfrom
andrewor14:delete-aqt
Open

Remove references to torchao's AffineQuantizedTensor#45321
andrewor14 wants to merge 2 commits intohuggingface:mainfrom
andrewor14:delete-aqt

Conversation

@andrewor14
Copy link
Copy Markdown
Contributor

Summary: TorchAO recently deprecated AffineQuantizedTensor and related classes (pytorch/ao#2752). These will be removed in the next release. We should remove references of these classes in transformers before then.

Test Plan:

python -m pytest -s -v tests/quantization/torchao_integration/test_torchao.py

@andrewor14
Copy link
Copy Markdown
Contributor Author

Hi @SunMarc can you help review this?

@jerryzh168
Copy link
Copy Markdown
Contributor

also cc @MekkCyber

Summary: TorchAO recently deprecated AffineQuantizedTensor and
related classes (pytorch/ao#2752). These will be removed in the
next release. We should remove references of these classes in
transformers before then.

Test Plan:
```
python -m pytest -s -v tests/quantization/torchao_integration/test_torchao.py
```
andrewor14 added a commit to andrewor14/peft that referenced this pull request Apr 8, 2026
**Summary:** TorchAO recently deprecated AffineQuantizedTensor
and related classes (pytorch/ao#2752). These will be removed
in the next release. We should remove references of these
classes in peft before then.

Similar to:
- huggingface/diffusers#13405
- huggingface/transformers#45321

**Test Plan:** CI
Copy link
Copy Markdown
Member

@SunMarc SunMarc left a comment

Choose a reason for hiding this comment

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

Sounds good to me thanks !

@github-actions
Copy link
Copy Markdown
Contributor

github-actions bot commented Apr 9, 2026

[For maintainers] Suggested jobs to run (before merge)

run-slow: torchao_integration

@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@SunMarc SunMarc enabled auto-merge April 9, 2026 13:24
@SunMarc SunMarc added this pull request to the merge queue Apr 9, 2026
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Apr 9, 2026
@SunMarc SunMarc added this pull request to the merge queue Apr 9, 2026
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Apr 9, 2026
@SunMarc SunMarc added this pull request to the merge queue Apr 9, 2026
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Apr 9, 2026
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