From 25bf0fa99e702f38fad1e2f76c1893c564c3404b Mon Sep 17 00:00:00 2001 From: peterjc123 Date: Fri, 13 Sep 2024 23:17:34 +0800 Subject: [PATCH] Update quantizer.py --- tinynn/graph/quantization/quantizer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tinynn/graph/quantization/quantizer.py b/tinynn/graph/quantization/quantizer.py index a19975ae..460f9b1a 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 = {}