+ "ݳj*+^RIt^RIt^RIt^RIt^RIt^RIt^RIHtHt^RI t .R Ot ^t !RR]P4t !RR]P4tRRlt0R mtR R ltR#) N)AnyDict AOBaseConfigc6a]tRt^toRt]tV3RltRtVt R#)ra If a workflow config inherits from this then `quantize_` knows how to a apply it to a model. For example:: # user facing code class WorkflowFooConfig(AOBaseConfig): ... # configuration for workflow `Foo` is defined here bar = 'baz' # non user facing code @register_quantize_module_handler(WorkflowFooConfig) def _transform( mod: torch.nn.Module, config: WorkflowFooConfig, ) -> torch.nn.Module: # the transform is implemented here, usually a tensor sublass # weight swap or a module swap ... # then, the user calls `quantize_` with a config, and `_transform` is called # under the hood by `quantize_. c&<V^8dQh/S[;R&#)version)int)format __classdict__s"T/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torchao/core/config.py __annotate__AOBaseConfig.__annotate__sP#QN) __name__ __module__ __qualname____firstlineno____doc___DEFAULT_VERSIONr __annotate_func____static_attributes____classdictcell__)r s@r rrs0$GQrc<aa]tRt^FtoRtV3RltRtRtVtV;t #)ConfigJSONEncoderz,Custom JSON encoder for AOBaseConfig objectsc r<\V\4'd/pVPP4F:wr4VP R4'dKVR8wgK'VP V4W#&K< RVP PR\VR\4RV/#\VR4'd\VR4'dnVP4pVP4UUu/uFwr4W0P V4bK pppRVP PR\VR\4RV/#\P!V4'd\V\4'g/p\P!V4FFpVP R8wgKVP \WP 44W'P &KH RVP PR\VR\4RV/#\VR4'dMVP"R8Xd<\V\4'd&RR R\%V4P'R 4R,/#\VR 4'dR VP P9doRVP PRVPP4UUu/uF/wr4VP R4'dKW0P V4bK1 upp/#\V\(P*4'd&RVP P RVP /#\V\,P.4'd&RR R\%V4P'R 4R,/#\V\04'd!VUu.uFqP V4NK up#\V\24'd\5R V 24h\V\64'd4VP4UUu/uFwr4W0P V4bK upp#\8S V`uV4#uuppiuuppiuupiuuppi)_r _type_version_data_fields_asdictrtorch torch.dtype. __class__LayoutiTuples will be serialized as List in JSON, so we recommend to use Lists instead to avoid surprises. got: ) isinstancer__dict__items startswith encode_valuer'rgetattrrhasattrr# dataclasses is_dataclasstypefieldsnamerstrsplitenumEnumr$dtypelisttupleNotImplementedErrordictsuperdefault) selfo data_dictkv asdict_dataprocessed_datafitemr's && r rAConfigJSONEncoder.defaultIs a & &I ((*||C((Q)^#'#4#4Q#7IL+--GAy2BC   1i W y& & ))+KBMBSBSBUVBU$!a!2!21!55BUNV--GAy2BC   # #A & &z!T/B/BI ''*66Y&(,(9(9'!VV:L(MIff%+ --GAy2BC   1l # # (?JqRVDWDW]GSV\\#5Fr5JK K 1k " "x1;;3G3G'G-- ! 0 0 2 2<<,,A((++ 2  a # # 4 45H H a % %]GSV\\#5Fr5JK K a  89:%%d+: : 5 ! !%::;>  a  89 B A((++ B Bwq!!yW>";Cs$P",P( P( P.3P3c NVPV4pV# \dT#i;i)z+Helper method to recursively encode a value)rA TypeError)rBvalueresults&& r r/ConfigJSONEncoder.encode_values5 \\%(FM    s  $$r) rrrrrrAr/rr __classcell__)r'r s@@r rrFs6S"j  rrcRV^8dQhR\R\\\3,/#)rconfigreturn)rrr7r)r s"r rrs&AA<ADcNArc\V\4'g\R\V4 24h\P !\P !V\R74#)z Convert an AOBaseConfig instance to a dictionary suitable for serialization. Args: config: An instance of AOBaseConfig subclass Returns: Dict representation of the config z$Expected AOBaseConfig instance, got )cls)r+rrMr4jsonloadsdumpsr)rSs&r config_to_dictrZsE fl + +>tF|nMNN ::djj->? @@rcRV^8dQhR\\\3,R\/#)rdatarT)rr7rr)r s"r rrs*eNeN4S>eNleNrc "\V\4'g\R\V4 24hRV9gRV9d \ R4hVR,pVP R\ 4pVR,pVR8Xd^RIp\WC4#Rp\F%p\P!V4p\Wq4pM Vf'RP\4p\ R V R V 24h\VR \ 4p W)8wd \P !R V: R V : R24\V\4'g5\#V\$P&4'd \WS4#V!V4#W)8wdR V/p M/p VP)4EF+wr\V \4'dRV 9dRV 9d\+V 4W&K9\V \,4'dFV U u.uF4p \V \4'dRV 9dRV 9d \+V 4MT NK6 up W&K\V \.4'd\1RV 24h\V \4'dZV P)4UUu/uF7wrT\V\4'dRV9dRV9d \+V4MTbK9 uppW&EK'WV &EK. V!R/V B# \\3dEK>i;i Tu#;iuup iuuppi \2d!p\ RTP4 RT 24hRp?ii;i)a Create an AOBaseConfig subclass instance from a dictionary. Args: data: Dictionary containing serialized config data Returns: An instance of the appropriate AOBaseConfig subclass Raises: ValueError: If deserialization fails for other reasons zExpected dictionary, got rr!z;Input dictionary missing required '_type' or '_data' fieldsr r%Nz, zFailed to find class z in any of the allowed modules: r zXStored version is not the same as current default version of the config: stored_version=z, current_default_version=z&, please check the deprecation warningr)zFailed to create instance of z: r)r+r?rMr4 ValueErrorgetrr$r0ALLOWED_AO_MODULES importlib import_module ImportErrorAttributeErrorjoinwarningswarn issubclassr9r:r-config_from_dictr<r=r> Exceptionr)r\ type_pathstored_versionobj_datar$rV module_pathmoduleallowed_modules_strcurrent_default_versionrHkeyrNrJrErFes& r riris- dD ! !3DJ<@AAdgT1VWWW IXXj*:;NG}HM!u'' C)  ,,[9F&,C  * {"ii(:;#I;.NObNc d  &c96FG0 gXfWhiDlCkEEk l h % % c499 % %3) ) 8}$ 0#^4nn&  eT " "w%'7Gu torchao.dtypestorchao.quantizationtorchao.prototype.awqtorchao.prototype.parqtorchao.sparsity.sparse_apitorchao.prototype.mx_formatstorchao.prototype.parq.quanttorchao.prototype.smoothquanttorchao.prototype.quantization%torchao.quantization.quantize_.common(torchao.quantization.quantize_.workflows)abcr2r9rarWrftypingrrr$__all__rABCr JSONEncoderrrZr`rirrr rsi      ($377($Ve((ePA" eNr