pin requirements

10 months ago · f450611ce3
1 changed files with 1 additions and 3 deletions
--- 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