Update requirements.txt

requirements.txt

@@ -1,7 +1,10 @@
 # ----------  PyTorch wheels that bundle the CUDA-12.1 runtime ----------
 --extra-index-url https://download.pytorch.org/whl/cu121
-torch==2.2.2+cu121
-torchvision==0.17.2+cu121        # must match Torch’s build
+# torch==2.2.2+cu121
+# torchvision==0.17.2+cu121        # must match Torch’s build
+
+torch
+torchvision
 
 # ----------  Extra index that hosts the PyTorch3D binary wheels ----------
 --extra-index-url https://miropsota.github.io/torch_packages_builder