-
Notifications
You must be signed in to change notification settings - Fork 380
Closed
Labels
enhancementNew feature or requestNew feature or request
Description
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
msaroufim, gau-nernst and jerryzh168
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or request