JiantaoLin
commited on
Commit
·
90eb8de
1
Parent(s):
ec44d03
new
Browse files
app.py
CHANGED
|
@@ -143,7 +143,7 @@ def save_py3dmesh_with_trimesh_fast(meshes, save_glb_path=TEMP_MESH_ADDRESS, app
|
|
| 143 |
print(f"saving to {save_glb_path}")
|
| 144 |
#
|
| 145 |
#
|
| 146 |
-
|
| 147 |
def text_to_detailed(prompt, seed=None):
|
| 148 |
# print(torch.cuda.is_available())
|
| 149 |
# print(f"Before text_to_detailed: {torch.cuda.memory_allocated() / 1024**3} GB")
|
|
|
|
| 143 |
print(f"saving to {save_glb_path}")
|
| 144 |
#
|
| 145 |
#
|
| 146 |
+
@spaces.GPU
|
| 147 |
def text_to_detailed(prompt, seed=None):
|
| 148 |
# print(torch.cuda.is_available())
|
| 149 |
# print(f"Before text_to_detailed: {torch.cuda.memory_allocated() / 1024**3} GB")
|