From 518ab83a328ecbf57e7d63ec370e15dfa4336598 Mon Sep 17 00:00:00 2001
From: Bingxin <bingke@ethz.ch>
Date: Tue, 20 Aug 2024 18:10:32 +0200
Subject: [PATCH] [ENV] update torch version (security)

---
 environment.yaml | 2 +-
 requirements.txt | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/environment.yaml b/environment.yaml
index 77920f7..9c5d44e 100644
--- 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
index 33cc65a..02e2cb4 100644
--- 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
\ No newline at end of file