JiantaoLin commited on
Commit
73a0140
·
1 Parent(s): 8c14434
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -187,6 +187,7 @@ def image2mesh_preprocess_(input_image_, seed, use_mv_rgb=True):
187
 
188
  @spaces.GPU
189
  def image2mesh_main_(reference_3d_bundle_image, caption, seed, strength1=0.5, strength2=0.95, enable_redux=True, use_controlnet=True, if_video=True):
 
190
  global mesh_cache
191
  seed = int(seed) if seed is not None else None
192
 
 
187
 
188
  @spaces.GPU
189
  def image2mesh_main_(reference_3d_bundle_image, caption, seed, strength1=0.5, strength2=0.95, enable_redux=True, use_controlnet=True, if_video=True):
190
+ subprocess.run(['nvidia-smi'])
191
  global mesh_cache
192
  seed = int(seed) if seed is not None else None
193