Accelerate-Tests

- speed up a test with a smaller model
- add torch to reqs

test/requirements.in

@@ -15,3 +15,4 @@ pytest-cov
 black
 isort
 flake8
+torch

test/requirements.txt

@@ -398,6 +398,8 @@ tomli==2.0.0
     #   black
     #   coverage
     #   pytest
+torch==1.10.2
+    # via -r requirements.in
 tqdm==4.62.3
     # via
     #   huggingface-hub
@@ -415,6 +417,7 @@ typing-extensions==4.0.1
     #   black
     #   huggingface-hub
     #   tensorflow
+    #   torch
 urllib3[secure]==1.26.8
     # via
     #   dulwich

test/test_external.py

@@ -244,10 +244,9 @@ class TestLoadInterface(unittest.TestCase):
 
 class TestLoadFromPipeline(unittest.TestCase):
     def test_question_answering(self):
-        pipe = transformers.pipeline(model="lysandre/tiny-vit-random")
+        pipe = transformers.pipeline(model="sshleifer/bart-tiny-random")
         output = pipe(
-            "My name is Sylvain and I work at Hugging Face in Brooklyn",
-            "Where do I work?",
+            "My name is Sylvain and I work at Hugging Face in Brooklyn"
         )
         self.assertIsNotNone(output)