LutaoJiang commited on
Commit
ed53610
Β·
1 Parent(s): 9cc3979
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -170,7 +170,7 @@ def save_py3dmesh_with_trimesh_fast(meshes, save_glb_path=TEMP_MESH_ADDRESS, app
170
  fix_vert_color_glb(save_glb_path)
171
  print(f"saving to {save_glb_path}")
172
 
173
- @spaces.GPU
174
  def text_to_detailed(prompt, seed=None):
175
  # test nvdiffrast
176
  subprocess.run(
@@ -187,7 +187,6 @@ def text_to_detailed(prompt, seed=None):
187
  subprocess.run(['nvidia-smi']) # Test if CUDA is available
188
  print(f"torch.cuda.is_available:{torch.cuda.is_available()}")
189
  print("Device count:", torch.cuda.device_count())
190
- subprocess.run(['nvidia-smi'])
191
  import nvdiffrast.torch as dr
192
  dr.RasterizeCudaContext()
193
  print("nvdiffrast initialized successfully")
 
170
  fix_vert_color_glb(save_glb_path)
171
  print(f"saving to {save_glb_path}")
172
 
173
+ @spaces.GPU(duration=120)
174
  def text_to_detailed(prompt, seed=None):
175
  # test nvdiffrast
176
  subprocess.run(
 
187
  subprocess.run(['nvidia-smi']) # Test if CUDA is available
188
  print(f"torch.cuda.is_available:{torch.cuda.is_available()}")
189
  print("Device count:", torch.cuda.device_count())
 
190
  import nvdiffrast.torch as dr
191
  dr.RasterizeCudaContext()
192
  print("nvdiffrast initialized successfully")