Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
b12ec62
1
Parent(s):
9e32a0a
add
Browse files
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/
|
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
|