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