CharlieAmalet commited on
Commit
e158176
·
verified ·
1 Parent(s): f29e15c

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -4,7 +4,7 @@ scipy==1.11.4
4
  pydantic==2.5.3
5
  timm==0.6.12
6
  trimesh==4.0.8
7
- --extra-index-url https://download.pytorch.org/whl/cu118
8
  torch==2.0.0
9
  torchvision==0.15.1
10
  torchaudio==2.0.1
 
4
  pydantic==2.5.3
5
  timm==0.6.12
6
  trimesh==4.0.8
7
+ --extra-index-url https://download.pytorch.org/whl/rocm5.4.2
8
  torch==2.0.0
9
  torchvision==0.15.1
10
  torchaudio==2.0.1