[ENV] update torch version (security)

prs-eth · Aug 20, 2024 · 518ab83 · 518ab83
1 parent a4187ae
commit 518ab83
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/environment.yaml b/environment.yaml
@@ -9,7 +9,7 @@ dependencies:
   - matplotlib>=3.8.2
   - scipy>=1.11.4
   - python=3.10.12
-  - pytorch=2.0.1
+  - pytorch=2.2.1
   - pytorch-cuda=11.7
   - torchvision=0.15.2
   - pip:

diff --git a/requirements.txt b/requirements.txt
@@ -2,6 +2,6 @@ accelerate>=0.22.0
 diffusers>=0.25.0
 matplotlib
 scipy
-torch==2.0.1
+torch==2.2.0
 torchvision==0.15.2
 transformers>=4.32.1