[ENV] fix env conflix #121

markkua · markkua · commit 62413d56099d · 2024-09-17T20:03:43.000+02:00
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
