qitaoz commited on
Commit
367570f
·
verified ·
1 Parent(s): d7bd894

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +9 -9
requirements.txt CHANGED
@@ -1,15 +1,15 @@
1
  # ---------- PyTorch wheels that bundle the CUDA-12.1 runtime ----------
2
- # --extra-index-url https://download.pytorch.org/whl/cu121
3
- # torch==2.2.2+cu121
4
- # torchvision==0.17.2+cu121 # must match Torch’s build
5
 
6
- torch==2.4.0
7
- torchvision==0.19.0
8
 
9
  # ---------- Extra index that hosts the PyTorch3D binary wheels ----------
10
  --extra-index-url https://miropsota.github.io/torch_packages_builder
11
- # pytorch3d==0.7.8+pt2.2.2cu121 # ← EXACT tag; NO “git+…” line
12
- pytorch3d==0.7.7+pt2.4.0cu121 # ← EXACT tag; NO “git+…” line
13
 
14
  black
15
  einops
@@ -43,5 +43,5 @@ accelerate==0.32.1
43
  gradio==4.44.1
44
  pydantic==2.10.6
45
  gdown
46
- # xformers==0.0.25.post1
47
- xformers==0.0.27.post1
 
1
  # ---------- PyTorch wheels that bundle the CUDA-12.1 runtime ----------
2
+ --extra-index-url https://download.pytorch.org/whl/cu121
3
+ torch==2.2.2+cu121
4
+ torchvision==0.17.2+cu121
5
 
6
+ # torch==2.4.0
7
+ # torchvision==0.19.0
8
 
9
  # ---------- Extra index that hosts the PyTorch3D binary wheels ----------
10
  --extra-index-url https://miropsota.github.io/torch_packages_builder
11
+ pytorch3d==0.7.8+pt2.2.2cu121
12
+ # pytorch3d==0.7.7+pt2.4.0cu121
13
 
14
  black
15
  einops
 
43
  gradio==4.44.1
44
  pydantic==2.10.6
45
  gdown
46
+ xformers==0.0.25.post1
47
+ # xformers==0.0.27.post1