envs

requirements.txt

@@ -214,10 +214,9 @@ scikit-learn==1.1.3
 scipy==1.10.1
 six==1.16.0
 toolz==0.12.1
-torch==2.3.0+cu121
-torchaudio==2.3.0+cu121
-torchmetrics==1.4.0.post0
-torchvision==0.18.0+cu121
+torch==2.3.0
+torchaudio==2.3.0
+torchvision==0.18.0
 tqdm==4.65.0
 transformers==4.32.1
 transforms3d==0.4.1