We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b0dc0de commit c4de198Copy full SHA for c4de198
examples/pytorch/nlp/huggingface_models/question-answering/quantization/ptq_static/ipex/run_qa.py
@@ -687,7 +687,10 @@ def eval_func(model):
687
example_inputs = get_example_inputs(model, eval_dataloader)
688
model = ipex.optimize(model)
689
with torch.no_grad():
690
- model = torch.jit.trace(model, example_kwarg_inputs=example_inputs, strict=False)
+ if isinstance(example_inputs, dict):
691
+ model = torch.jit.trace(model, example_kwarg_inputs=example_inputs, strict=False)
692
+ else:
693
+ model = torch.jit.trace(model, example_inputs, strict=False)
694
model = torch.jit.freeze(model)
695
696
if model_args.benchmark or model_args.accuracy_only:
0 commit comments