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