[ENV] fix env conflix #121

prs-eth · Sep 17, 2024 · 62413d5 · 62413d5
1 parent 518ab83
commit 62413d5
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 10 deletions.
diff --git a/README.md b/README.md
@@ -61,10 +61,7 @@ paper and inference code (this repository).
 
 The inference code was tested on:
 
-- Ubuntu 22.04 LTS, Python 3.10.12,  CUDA 11.7, GeForce RTX 3090 (pip, Mamba)
-- CentOS Linux 7, Python 3.10.4, CUDA 11.7, GeForce RTX 4090 (pip)
-- Windows 11 22H2, Python 3.10.12, CUDA 12.3, GeForce RTX 3080 (Mamba)
-- MacOS 14.2, Python 3.10.12, M1 16G (pip)
+- Ubuntu 22.04 LTS, Python 3.10.12,  CUDA 11.7, GeForce RTX 3090 (pip)
 
 ### 🪧 A Note for Windows users
 

diff --git a/environment.yaml b/environment.yaml
@@ -9,8 +9,8 @@ dependencies:
   - matplotlib>=3.8.2
   - scipy>=1.11.4
   - python=3.10.12
-  - pytorch=2.2.1
+  - pytorch=2.4.0
   - pytorch-cuda=11.7
-  - torchvision=0.15.2
+  - torchvision=0.19.0
   - pip:
-    - transformers==4.32.1
+    - transformers>=4.32.1
diff --git a/requirements++.txt b/requirements++.txt
@@ -2,5 +2,5 @@ h5py
 opencv-python
 tensorboard
 wandb
-xformers==0.0.21
+xformers==0.0.27.post2
 
diff --git a/requirements.txt b/requirements.txt
@@ -2,6 +2,6 @@ accelerate>=0.22.0
 diffusers>=0.25.0
 matplotlib
 scipy
-torch==2.2.0
-torchvision==0.15.2
+torch==2.4.0
+torchvision==0.19.0
 transformers>=4.32.1