diff --git a/extension/llm/export/builder.py b/extension/llm/export/builder.py index 3e98748969..47ad30e939 100644 --- a/extension/llm/export/builder.py +++ b/extension/llm/export/builder.py @@ -360,7 +360,10 @@ def pt2e_quantize(self, quantizers: Optional[List[Quantizer]]) -> "LLMEdgeManage logging.info( "No calibration provided, using dummy input to calibrate..." ) - m(*self.example_inputs) + if self.example_kwarg_inputs: + m(*self.example_inputs, **self.example_kwarg_inputs) + else: + m(*self.example_inputs) m = convert_pt2e(m) DuplicateDynamicQuantChainPass()(m) self.pre_autograd_graph_module = m