add reqs failing test

sdan · Apr 17, 2024 · d40cdf0 · d40cdf0
1 parent 751eeac
commit d40cdf0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,5 @@
+torch==2.2.2
+transformers==4.39.0
 numpy
 PyPDF2
 docx2txt
@@ -8,5 +10,3 @@ huggingface_hub
 tiktoken
 onnxruntime==1.17.1
 tokenizers==0.15.2
-torch==2.2.2
-transformers==4.39.0
diff --git a/setup.py b/setup.py
@@ -18,7 +18,8 @@
         'beautifulsoup4',
         'transformers',
         'huggingface_hub',
-        'tiktoken'
+        'tiktoken',
+        'torch',
     ],
     extras_require={
         'ocr': ['surya-ocr-vlite']