h+SSKrSSKrSSKrSSKrSSKrSSKrSSKJrJr SSK r /SQr Sr "SS\R5r "SS\R5rS \ S \\\44S jr1S krS \\\4S \ 4Sjrg)N)AnyDict) AOBaseConfigconfig_from_dictconfig_to_dictALLOWED_AO_MODULESc*\rSrSr%Sr\r\\S'Sr g)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_. versionN) __name__ __module__ __qualname____firstlineno____doc___DEFAULT_VERSIONr int__annotations____static_attributes__r M/home/james-whalen/.local/lib/python3.13/site-packages/torchao/core/config.pyrrs0$GS#rrc2^\rSrSrSrU4SjrSrSrU=r$)ConfigJSONEncoderFz,Custom JSON encoder for AOBaseConfig objectsc 0>[U[5(a0nURR5H8up4UR S5(aMUS:wdM%UR U5X#'M: UR R[US[5US.$[US5(a}[US5(alUR5nUR5VVs0sHup4X0R U5_M nnnUR R[US[5US.$[R"U5(a[U[5(d0n[R"U5HEnUR S:wdMUR [XR 55X'R 'MG UR R[US[5US.$[US5(aEUR"S:Xa5[U[5(a S[%U5R'S 5S S .$[US 5(aS UR R;amUR RURR5VVs0sH.up4UR S5(aMX0R U5_M0 snnS .$[U[(R*5(a$UR RUR S .$[U[,R.5(a S[%U5R'S 5S S .$[U[05(a!UVs/sHoR U5PM sn$[U[25(a[5SU35e[U[65(a4UR5VVs0sHup4X0R U5_M snn$[8T U]uU5$s snnfs snnfs snfs snnf)N_r )_type_version_data_fields_asdictrtorch torch.dtype.)rr __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 rr@ConfigJSONEncoder.defaultIsi a & &I ((*||C((Q)^#'#4#4Q#7IL+--#Ay2BC"   1i W y& & ))+KBMBSBSBUVBU$!a!2!21!55BUNV--#Ay2BC'   # #A & &z!T/B/BI ''*66Y&(,(9(9'!VV:L(MIff%+ --#Ay2BC"   1l # # (?JqRVDWDW*SV\\#5Fr5JK K 1k " "x1;;3G3G'G--!" 0 0 2 2<<,,A((++ 2  a # # ! 4 45H H a % %*SV\\#5Fr5JK K a  89:%%d+: : 5 ! !%::;>  a  89 B A((++ B Bwq!!yW>";CsPP3P)P PcLURU5nU$![a U$f=f)z+Helper method to recursively encode a value)r@ TypeError)rAvalueresults rr.ConfigJSONEncoder.encode_values5 \\%(FM    s  ##r ) rrrrrr@r.r __classcell__)r's@rrrFs6S"j  rrconfigreturnc[U[5(d[S[U535e[R "[R "U[S95$)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*rrLr3jsonloadsdumpsr)rQs rrrsE fl + +>tF|nMNN ::djj->? @@r> torchao.dtypestorchao.quantizationtorchao.prototype.awqtorchao.prototype.parqtorchao.sparsity.sparse_apitorchao.prototype.mx_formatstorchao.prototype.parq.quanttorchao.prototype.quantization%torchao.quantization.quantize_.common(torchao.quantization.quantize_.workflowsdatac [U[5(d[S[U535eSU;dSU;a [ S5eUSnUR S[ 5nUSnUS:XaSSKn[XC5$Sn[H%n[R"U5n[Xq5n O Uc&S R[5n[ S US U35e[US [ 5n X):wa[R "S U<SU <S35 [U[5(d3[#U[$R&5(a [XS5$U"U5$X):waS U0n O0n UR)5GH!up[U [5(aSU ;aSU ;a[+U 5X'M6[U [,5(aCU V s/sH1n [U [5(aSU ;aSU ;a [+U 5OU PM3 sn X'M[U [.5(a[1SU 35e[U [5(aWU R)5VVs0sH4upU[U[5(aSU;aSU;a [+U5OU_M6 snnX'GMXU 'GM$ U"S0U D6$![[4a GM.f=f! Us$=fs sn fs snnf![2a n[ SUR4SU35eSnAff=f)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' fieldsrr$rNz, 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>rLr3 ValueErrorgetrr#r/r importlib import_module ImportErrorAttributeErrorjoinwarningswarn issubclassr8r9r,rr;r<r= Exceptionr)rb type_pathstored_versionobj_datar#rT module_pathmoduleallowed_modules_strcurrent_default_versionrGkeyrMrIrDrEes rrrs- 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%'7Gur}s      ($377($Ve((ePA<ADcNA" eN4S>eNleNr