YiftachEde commited on
Commit
b12ec62
·
1 Parent(s): 9e32a0a
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import os
2
  # this is a HF Spaces specific hack, as
3
  # (i) building pytorch3d with GPU support is a bit tricky here
4
  # (ii) installing the wheel via requirements.txt breaks ZeroGPU
5
- os.system("pip install pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt221/download.html")
6
 
7
  import torch
8
  import torch.nn as nn
 
2
  # this is a HF Spaces specific hack, as
3
  # (i) building pytorch3d with GPU support is a bit tricky here
4
  # (ii) installing the wheel via requirements.txt breaks ZeroGPU
5
+ os.system("pip install pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu118_pyt240/download.html")
6
 
7
  import torch
8
  import torch.nn as nn