wip: pin sentence-transformers

GT4SD · Mar 7, 2024 · b8b9c60 · b8b9c60
1 parent 17ae3b4
commit b8b9c60
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/conda.yml b/conda.yml
@@ -7,10 +7,9 @@ channels:
 dependencies:
  - python>=3.7,<3.9
  - pip>=19.1,<20.3
- - cudatoolkit=10.2
  - pytorch>=1.0,<=1.12.1
  - cpuonly
- - pytorch-scatter<=2.0.9
+ - pytorch-scatter<=2.0.9=*cu102*
  - pip:
  - -r requirements.txt
  - -r cpu_requirements.txt

diff --git a/conda_gpu.yml b/conda_gpu.yml
@@ -8,7 +8,7 @@ dependencies:
  - python>=3.7,<3.9
  - pip>=19.1,<20.3
  - pytorch>=1.0,<=1.12.1=*cu*
- - pytorch-scatter<=2.0.9,cudatoolkit=10.2
+ - pytorch-scatter<=2.0.9=*cu102*
  - torchvision<=0.13.1=*cu*
  - torchaudio<=0.12.1=*cu*
  - pip:

diff --git a/requirements.txt b/requirements.txt
@@ -32,6 +32,7 @@ scikit-learn>=1.0.0,<1.3.0
 scikit-optimize>=0.8.1
 scipy>=1.0.0
 sentencepiece>=0.1.95
+sentence_transformers>1.0,<=2.2.2
 sympy>=1.10.1
 tables>=3.7.0
 tape-proteins>=0.4