Skip to content

Commit ee032ac

Browse files
authored
Update examples/openvino/aot_openvino_compiler.py
1 parent 4f8ca40 commit ee032ac

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

examples/openvino/aot_openvino_compiler.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,8 @@ def main(
187187
batch_size = calibration_dataset.batch_size
188188
subset_size = (subset_size // batch_size) + int(subset_size % batch_size > 0)
189189

190-
transform_fn = lambda x: x[0]
190+
def transform_fn(x):
191+
return x[0]
191192
quantized_model = quantize_model(
192193
aten_dialect.module(),
193194
calibration_dataset,

0 commit comments

Comments
 (0)