Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
3951932
1
Parent(s):
6b78472
add
Browse files
app.py
CHANGED
|
@@ -2,6 +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 |
|
| 6 |
# Use the dynamic approach from PyTorch3D documentation to determine the correct wheel
|
| 7 |
import sys
|
|
@@ -46,7 +47,6 @@ from shap_e.diffusion.sample import sample_latents
|
|
| 46 |
from shap_e.diffusion.gaussian_diffusion import diffusion_from_config
|
| 47 |
from shap_e.models.download import load_model, load_config
|
| 48 |
from shap_e.util.notebooks import create_pan_cameras, decode_latent_images
|
| 49 |
-
import spaces
|
| 50 |
from shap_e.models.nn.camera import DifferentiableCameraBatch, DifferentiableProjectiveCamera
|
| 51 |
import math
|
| 52 |
import time
|
|
|
|
| 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 |
+
import spaces
|
| 6 |
|
| 7 |
# Use the dynamic approach from PyTorch3D documentation to determine the correct wheel
|
| 8 |
import sys
|
|
|
|
| 47 |
from shap_e.diffusion.gaussian_diffusion import diffusion_from_config
|
| 48 |
from shap_e.models.download import load_model, load_config
|
| 49 |
from shap_e.util.notebooks import create_pan_cameras, decode_latent_images
|
|
|
|
| 50 |
from shap_e.models.nn.camera import DifferentiableCameraBatch, DifferentiableProjectiveCamera
|
| 51 |
import math
|
| 52 |
import time
|