combine workflows

huggingface · Feb 27, 2024 · 52119bb · 52119bb
1 parent d822b2d
commit 52119bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/vit/test_training_vit.py b/tests/models/vit/test_training_vit.py
@@ -8,7 +8,7 @@
 # TODO: Find a way to get the installed `transformers` directory.
 SRC_DIRS = [
     # os.path.join("/transformers/examples/pytorch", dirname)
-    os.path.join("/usr/local/lib/python3.10/dist-packages/transformers", dirname)
+    os.path.join("/usr/local/lib/python3.10/dist-packages/transformers/examples/pytorch", dirname)
     for dirname in [
         "image-classification",
     ]