diff --git a/environment.yml b/environment.yml
index 4250d83..18fae2d 100755
--- 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