diff --git a/tinynn/graph/quantization/quantizer.py b/tinynn/graph/quantization/quantizer.py index a19975a..460f9b1 100644 --- a/tinynn/graph/quantization/quantizer.py +++ b/tinynn/graph/quantization/quantizer.py @@ -264,7 +264,6 @@ if hasattr(nn, 'RMSNorm'): UNSUPPORTED_PYTORCH_QUANTIZATION_OP_LIST.update({nn.RMSNorm: None}) - FUNCTIONAL_MODULE_MAPPING.update({'rms_norm': nn.RMSNorm}) # Processed QAT fuse rules processed_qat_rules = {}