From f450611ce3d9070da2fbb0c737a41d34038b2ba3 Mon Sep 17 00:00:00 2001
From: Michael Pilosov <consistentbayes@gmail.com>
Date: Sat, 27 Jan 2024 23:33:05 -0700
Subject: [PATCH] pin requirements

---
 requirements.txt | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 25fb98b..61561cc 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -181,10 +181,8 @@ terminado==0.18.0
 threadpoolctl==3.2.0
 tinycss2==1.2.1
 tomli==2.0.1
-torch==2.1.1+cu121
-torchaudio==2.1.1+cu121
+torch==2.1.1+cu118
 torchmetrics==1.3.0.post0
-torchvision==0.16.1+cu121
 tornado==6.4
 tqdm==4.66.1
 traitlets==5.14.1