JiantaoLin
commited on
Commit
·
ec44d03
1
Parent(s):
da20481
new
Browse files
app.py
CHANGED
@@ -144,9 +144,9 @@ def save_py3dmesh_with_trimesh_fast(meshes, save_glb_path=TEMP_MESH_ADDRESS, app
|
|
144 |
#
|
145 |
#
|
146 |
|
147 |
-
@spaces.GPU
|
148 |
def text_to_detailed(prompt, seed=None):
|
149 |
-
print(
|
|
|
150 |
return k3d_wrapper.get_detailed_prompt(prompt, seed)
|
151 |
|
152 |
@spaces.GPU
|
|
|
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")
|
150 |
return k3d_wrapper.get_detailed_prompt(prompt, seed)
|
151 |
|
152 |
@spaces.GPU
|