diff --git a/src/brevitas_examples/common/generative/quantize.py b/src/brevitas_examples/common/generative/quantize.py index f3a1c4add..87392a83f 100644 --- a/src/brevitas_examples/common/generative/quantize.py +++ b/src/brevitas_examples/common/generative/quantize.py @@ -181,7 +181,8 @@ 'dynamic': { 'float_scale': { 'stats': { - 'per_group': Fp8e4m3DynamicActPerGroupFloat}}}, + 'per_group': { + 'sym': Fp8e4m3DynamicActPerGroupFloat}}}}, 'no_scale': { 'sym': Fp8e4m3Act,}}, 'float_ocp': {