update requirements

probabl-ai · Dec 5, 2023 · 742d689 · 742d689
1 parent 7187970
commit 742d689
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/environment.yml b/environment.yml
@@ -10,9 +10,9 @@ dependencies:
   - pytest
   - pytest-cov
   - pytorch
-  - transformers
   - scikit-learn
   - sentence-transformers
+  - transformers
   - pip:
     # We use return_generator option that will be shipped in 1.4
     - git+https://github.com/joblib/joblib.git@master
diff --git a/requirements.txt b/requirements.txt
@@ -3,5 +3,9 @@ langchain
 pytest
 pytest-cov
 scikit-learn
+sentence-transformers
+transformers
 # We use return_generator option that will be shipped in 1.4
 joblib @ git+https://github.com/joblib/joblib.git@master
+--index-url https://download.pytorch.org/whl/cpu
+torch