From 94198119a1a9078b5e11a7ee34dbba36f3583fe8 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Wed, 21 Aug 2024 18:58:27 +0100 Subject: [PATCH] update --- src/brevitas_examples/llm/llm_quant/export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/brevitas_examples/llm/llm_quant/export.py b/src/brevitas_examples/llm/llm_quant/export.py index 0998fd417..3d80fb644 100644 --- a/src/brevitas_examples/llm/llm_quant/export.py +++ b/src/brevitas_examples/llm/llm_quant/export.py @@ -24,8 +24,8 @@ from brevitas.function.ops import max_int from brevitas.function.ops import min_int from brevitas.nn import QuantLinear -from brevitas.proxy.parameter_quant import WeightQuantProxyFromInjector from brevitas.proxy.groupwise_int_parameter_quant import GroupwiseWeightQuantProxyFromInjector +from brevitas.proxy.parameter_quant import WeightQuantProxyFromInjector # TODO: Improve Groupwise export