Skip to content

Commit 9fe1e84

Browse files
authored
Deal with none
1 parent 18638eb commit 9fe1e84

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

extension/llm/export/builder.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -357,7 +357,10 @@ def pt2e_quantize(self, quantizers: Optional[List[Quantizer]]) -> "LLMEdgeManage
357357
logging.info(
358358
"No calibration provided, using dummy input to calibrate..."
359359
)
360-
m(*self.example_inputs, **self.example_kwarg_inputs)
360+
if self.example_kwarg_inputs:
361+
m(*self.example_inputs, **self.example_kwarg_inputs)
362+
else:
363+
m(*self.example_inputs)
361364
m = convert_pt2e(m)
362365
DuplicateDynamicQuantChainPass()(m)
363366
self.pre_autograd_graph_module = m

0 commit comments

Comments
 (0)