Skip to content

Commit 7597c29

Browse files
committed
udpate model test
1 parent 45dab01 commit 7597c29

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

.github/workflows/test_openvino_examples.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -44,4 +44,4 @@ jobs:
4444
4545
- name: Test examples
4646
run: |
47-
python examples/openvino/test_examples.py
47+
python -m pytest examples/openvino/test_examples.py

examples/openvino/test_examples.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ def test_image_classification(self):
7373
with tempfile.TemporaryDirectory() as tmp_dir:
7474
test_args = f"""
7575
run_image_classification.py
76-
--model_name_or_path nateraw/vit-base-beans
76+
--model_name_or_path hf-internal-testing/tiny-random-ViTModel
7777
--dataset_name beans
7878
--max_train_samples 10
7979
--max_eval_samples 2

0 commit comments

Comments
 (0)