🐳 chore: update requirements.txt

README.md

@@ -4,7 +4,7 @@ emoji: 💃
 colorFrom: blue
 colorTo: green
 sdk: gradio
-sdk_version: 5.20.1
+sdk_version: 5.25.2
 python_version: 3.10.13
 app_file: app.py
 pinned: true

requirements.txt

@@ -1,10 +1,10 @@
---extra-index-url https://download.pytorch.org/whl/cu117
-torch==2.0.1
-torchvision==0.15.2
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch==2.1.2
+torchvision==0.16.2
 
 einops
 timm
--f https://data.pyg.org/whl/torch-2.0.1+cu117.html
+-f https://data.pyg.org/whl/torch-2.1.2+cu121.html
 torch-cluster
 PyMCubes
 
@@ -23,6 +23,6 @@ tensorboard~=2.11.2
 tqdm
 trimesh
 # git+https://github.com/facebookresearch/pytorch3d.git@stable
-# pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu117_pyt201/download.html
+# pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt212/download.html
 --extra-index-url https://miropsota.github.io/torch_packages_builder
-pytorch3d==0.7.7+pt2.0.1cu117
+pytorch3d==0.7.7+pt2.1.2cu121