Merge pull request #17 from N-Hoffmann/main

Force downgrade of mkl
IGDRion · May 6, 2024 · 1c52259 · 1c52259
2 parents 4b0fde4 + 4415358
commit 1c52259
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/environment.yml b/environment.yml
@@ -3,7 +3,9 @@ dependencies:
   - conda-forge::git
 
   - conda-forge::python=3.10.4
-
+  - pytorch::pytorch=1.12.0
+  - pytorch::cpuonly
+
   - conda-forge::r-base=4.1
   - conda-forge::r-rcolorbrewer
   - conda-forge::r-tidyverse=1.3.2
@@ -21,5 +23,3 @@ dependencies:
       - git+https://github.com/igdrion/transforkmers.git
       - transformers==4.27.2
       - tokenizers==0.13.2
-      - torch==1.12.0
-      - mkl==2024.0