o ٷi@s~ddlZddlmZmZmZddlmZeeZdddddgdfdej j d ee d e d e d ee d eedej j fddZ dS)N)CallableOptionalList) init_loggeraoTembedderembedmodule quant_typebackendper_rowexclude_layers filter_fnreturncKspt|tjjs J|durd}td||dvr0ddlm}||f||||d|St d|d) Nfloat8_weight_onlyz,quant_type is not specified, using default: )rtorchao) quantize_ao)r r r rz backend: z is not supported now!) isinstancetorchnnModuleloggerwarninglowerrr ValueError)r r r r r rkwargsrrO/home/ubuntu/.local/lib/python3.10/site-packages/cache_dit/quantize/__init__.pyquantizes"    r)rtypingrrrcache_dit.loggerr__name__rrrstrboolrrrrrs4