JiantaoLin commited on
Commit
ec44d03
·
1 Parent(s): da20481
Files changed (1) hide show
  1. app.py +2 -2
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(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
 
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