From 3d1393c940d108b64a853549e96daa4e14b88402 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Tue, 21 Jan 2025 12:18:41 +0100 Subject: [PATCH] Update quantize.py --- src/brevitas_examples/common/generative/quantize.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/brevitas_examples/common/generative/quantize.py b/src/brevitas_examples/common/generative/quantize.py index 38cf24733..9cfd9b85c 100644 --- a/src/brevitas_examples/common/generative/quantize.py +++ b/src/brevitas_examples/common/generative/quantize.py @@ -16,7 +16,7 @@ from brevitas.core.zero_point import ParameterFromStatsFromParameterZeroPoint from brevitas.graph.quantize import layerwise_quantize from brevitas.quant.base import ParameterFromRuntimeZeroPoint -from brevitas.quant.experimental.float i +from brevitas.quant.experimental.float import Fp8e4m3Act from brevitas.quant.experimental.float import Fp8e4m3ActPerTensorFloat from brevitas.quant.experimental.float import Fp8e4m3WeightPerChannelFloat from brevitas.quant.experimental.float import Fp8e4m3WeightPerTensorFloat