Merge pull request #9 from shaneknapp/update-pytorch

bump pytorch to 2.5.1
berkeley-dsep-infra · Nov 19, 2024 · 6ffa8f2 · 6ffa8f2
2 parents 13dabe3 + a08de01
commit 6ffa8f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/environment.yml b/environment.yml
@@ -36,7 +36,7 @@ dependencies:
 - sqlalchemy==2.0.16
 - mlxtend==0.23.0
 # Spring 2024 data 100
-- pytorch==2.1.2
+- pytorch==2.5.1
 - cpuonly==2.0
 - transformers==4.37.1
 # Spring 2024 table demos