combine workflows

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