silentchen commited on
Commit
c0aff90
1 Parent(s): 5d94b0f

update space

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -178,7 +178,7 @@ def main():
178
  print(state)
179
  return out_gen_1, out_gen_2, out_gen_3, out_gen_4, edited_1, edited_2, edited_3, edited_4
180
 
181
- @spaces.GPU()
182
  @torch.no_grad()
183
  def generate_3d_with_shap_e(prompt, rand_seed, state):
184
  print("Check if I can use partial")
@@ -234,7 +234,7 @@ def main():
234
  del latents
235
  return mesh_path[0], mesh_path[1], mesh_path[2], mesh_path[3], state
236
 
237
- @spaces.GPU()
238
  @torch.no_grad()
239
  def _3d_editing(instruction, rand_seed, state):
240
  set_seed(rand_seed)
 
178
  print(state)
179
  return out_gen_1, out_gen_2, out_gen_3, out_gen_4, edited_1, edited_2, edited_3, edited_4
180
 
181
+ @spaces.GPU(duration=180)
182
  @torch.no_grad()
183
  def generate_3d_with_shap_e(prompt, rand_seed, state):
184
  print("Check if I can use partial")
 
234
  del latents
235
  return mesh_path[0], mesh_path[1], mesh_path[2], mesh_path[3], state
236
 
237
+ @spaces.GPU(duration=180)
238
  @torch.no_grad()
239
  def _3d_editing(instruction, rand_seed, state):
240
  set_seed(rand_seed)