udpate model test

huggingface · Mar 19, 2024 · 7597c29 · 7597c29
1 parent 45dab01
commit 7597c29
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/test_openvino_examples.yml b/.github/workflows/test_openvino_examples.yml
@@ -44,4 +44,4 @@ jobs:
 
     - name: Test examples
       run: |
-        python examples/openvino/test_examples.py
+        python -m pytest examples/openvino/test_examples.py
diff --git a/examples/openvino/test_examples.py b/examples/openvino/test_examples.py
@@ -73,7 +73,7 @@ def test_image_classification(self):
         with tempfile.TemporaryDirectory() as tmp_dir:
             test_args = f"""
                 run_image_classification.py
-                --model_name_or_path nateraw/vit-base-beans
+                --model_name_or_path hf-internal-testing/tiny-random-ViTModel
                 --dataset_name beans
                 --max_train_samples 10
                 --max_eval_samples 2