Update quantize.py

Xilinx · Jan 21, 2025 · 3d1393c · 3d1393c
1 parent a697c78
commit 3d1393c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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