From 36b8549f3ad8e4c1cc0ada16f61cde7cd6f974ca Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Thu, 16 Nov 2023 09:28:29 +1100
Subject: [PATCH] pin torch==2.1.0, torchvision=0.16.0

---
 pyproject.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 80421d9277..03a49d2e3b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -79,8 +79,8 @@ dependencies = [
   "semver~=3.0.1",
   "send2trash",
   "test-tube~=0.7.5",
-  "torch~=2.1.0",
-  "torchvision~=0.16",
+  "torch==2.1.0",
+  "torchvision==0.16.0",
   "torchmetrics~=0.11.0",
   "torchsde~=0.2.5",
   "transformers~=4.35.0",