Minor chnages

requirements.txt

@@ -95,12 +95,12 @@ tbb==2021.11.0
 threadpoolctl==3.4.0
 tomlkit==0.12.0
 toolz==0.12.1
-torch==2.3.0+cu121
+torch==2.3.0
 torch-lr-finder==0.2.1
-torchaudio==2.3.0+cu121
+torchaudio==2.3.0
 torchmetrics==1.3.2
 torchsummary==1.5.1
-torchvision==0.18.0+cu121
+torchvision==0.18.0
 tornado==6.4
 tqdm==4.66.2
 traitlets==5.14.3