Spaces:
Runtime error
Runtime error
Create shape.py
Browse files
shape.py
ADDED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from shap_e.diffusion.sample import sample_latents
|
2 |
+
from shap_e.diffusion.gaussian_diffusion import diffusion_from_config as diffusion_from_config_shape
|
3 |
+
from shap_e.models.download import load_model, load_config
|
4 |
+
from shap_e.util.notebooks import create_pan_cameras, decode_latent_images, gif_widget
|
5 |
+
from shap_e.util.notebooks import decode_latent_mesh
|
6 |
+
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
7 |
+
xm = load_model('transmitter', device=device)
|
8 |
+
model = load_model('text300M', device=device)
|
9 |
+
# model.load_state_dict(torch.load('./load/shapE_finetuned_with_330kdata.pth', map_location=device)['model_state_dict'])
|
10 |
+
diffusion = diffusion_from_config_shape(load_config('diffusion'))
|