From d822b2d68ba0d0c92a7d12611788253ecf51db11 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Tue, 27 Feb 2024 18:14:09 +0800 Subject: [PATCH] combine workflows --- tests/models/vit/test_training_vit.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/models/vit/test_training_vit.py b/tests/models/vit/test_training_vit.py index dcb7cb44..017c4022 100644 --- 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", ]