Spaces:

yejunliang23
/

ShapLLM-Omni

Running on Zero

yejunliang23 commited on 24 days ago

Commit

a18f018

verified ·

1 Parent(s): b44bdb9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -387,7 +387,7 @@ def _transform_messages(original_messages):
     return transformed_messages
 from trellis.models.sparse_structure_vqvae import VQVAE3D
-device       = torch.device("cpu")
 vqvae        = VQVAE3D(num_embeddings=8192)
 vqvae.eval()
 filepath = hf_hub_download(repo_id="yejunliang23/3DVQVAE",filename="3DVQVAE.bin")

     return transformed_messages
 from trellis.models.sparse_structure_vqvae import VQVAE3D
+device       = torch.device("cuda")
 vqvae        = VQVAE3D(num_embeddings=8192)
 vqvae.eval()
 filepath = hf_hub_download(repo_id="yejunliang23/3DVQVAE",filename="3DVQVAE.bin")