Add TorchAOHfQuantizer #32306
jerryzh168
changed the title [WIP] Add TorchAOHfQuantizer Add TorchAOHfQuantizer 1 year ago
msaroufim
approved these changes
on 2024-08-02
Add TorchAOHfQuantizer
5782f380
Fix a few issues
9b6b688b
style
85ea6fed
Added tests and addressed some comments about dtype conversion
fe4edddc
fix torch_dtype warning message
e2d0478c
SunMarc
approved these changes
on 2024-08-08
fix tests
a2284df6
style
25f132b6
TorchAOConfig -> TorchAoConfig
89ec2ce9
enable offload + fix memory with multi-gpu
91b4223e
update torchao version requirement to 0.4.0
597d8ad6
better comments
3c893a81
Merge branch 'torchao' of github.com:jerryzh168/transformers into tor…
78313eb5
add torch.compile to torchao README, add perf number link
130d2dfa
SunMarc
merged
78d78cdf
into main 1 year ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub