Skip to content

quantization api name consistency #375

@HDCharles

Description

@HDCharles

having the string to get a type of quantization and the constructor for that quantization as different text is super annoying.
https://github.com/pytorch/ao/blob/main/torchao/quantization/quant_api.py#L439-L443

quantize(model, int4wo()) ->works.......while quantize(model, "int4wo") -> error, wrong name

its a pain to have to remember which is which here, either both should work for the string or consolidate on a single one.

for that matter these also don't match

https://github.com/pytorch/ao/blob/main/torchao/quantization/quant_api.py#L310-L319

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions