Skip to content

Commit eb83a47

Browse files
Apply suggestions from code review
Co-authored-by: Alexander Suslov <[email protected]>
1 parent 00cd9a0 commit eb83a47

File tree

2 files changed

+3
-6
lines changed

2 files changed

+3
-6
lines changed

examples/openvino/aot/aot_openvino_compiler.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,6 @@
3030
from transformers import AutoModel
3131

3232
import nncf
33-
import nncf.experimental
34-
import nncf.experimental.torch
3533
from nncf.experimental.torch.fx.quantization.quantize_pt2e import quantize_pt2e
3634

3735

@@ -239,7 +237,7 @@ def transform(x):
239237
action="store_true",
240238
help="Enable model validation. --dataset argument is requred for the validation.",
241239
)
242-
parser.add_argument("--dataset", type=str, help="Path to the calibration dataset.")
240+
parser.add_argument("--dataset", type=str, help="Path to the validation dataset.")
243241
parser.add_argument(
244242
"--device",
245243
type=str,

examples/openvino/openvino_build.sh

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,13 +27,12 @@ main() {
2727

2828

2929
# Build the project
30-
cmake --build cmake-openvino-out --target install --config Release -j40
30+
cmake --build cmake-openvino-out --target install --config Release -j5
3131

3232
## Build example
3333
local example_dir=examples/openvino
3434
local example_build_dir="${build_dir}/${example_dir}"
3535
local cmake_prefix_path="${PWD}/${build_dir}/lib/cmake/ExecuTorch;${PWD}/${build_dir}/third-party/gflags;"
36-
3736
rm -rf "${example_build_dir}"
3837

3938
## OpenVINO original
@@ -42,7 +41,7 @@ main() {
4241
-B"${example_build_dir}" \
4342
$EXECUTORCH_ROOT/$example_dir
4443

45-
cmake --build "${example_build_dir}" -j40
44+
cmake --build "${example_build_dir}" -j5
4645

4746
# Switch back to the original directory
4847
cd - > /dev/null

0 commit comments

Comments
 (0)