diff --git a/ai_edge_quantizer/default_policy.py b/ai_edge_quantizer/default_policy.py index d803539..36fdc03 100644 --- a/ai_edge_quantizer/default_policy.py +++ b/ai_edge_quantizer/default_policy.py @@ -208,7 +208,7 @@ "EMBEDDING_LOOKUP", "FULLY_CONNECTED" ], - "weightonly_wi4_afp32": ["BATCH_MATMUL", "FULLY_CONNECTED"] + "weightonly_wi4_afp32": ["BATCH_MATMUL", "FULLY_CONNECTED", "EMBEDDING_LOOKUP"] } } """