soujanyaporia commited on
Commit
70761cd
·
verified ·
1 Parent(s): 33c4408

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -250,7 +250,7 @@ pipe = TangoPipeline(vae=tango.vae,
250
  )
251
 
252
 
253
- @spaces.GPU(duration=180)
254
  def gradio_generate(prompt, output_format, steps, guidance):
255
  output_wave = pipe(prompt,steps,guidance) ## Using pipeliine automatically uses flash attention for torch2.0 above
256
  #output_wave = tango.generate(prompt, steps, guidance)
 
250
  )
251
 
252
 
253
+ @spaces.GPU(duration=120)
254
  def gradio_generate(prompt, output_format, steps, guidance):
255
  output_wave = pipe(prompt,steps,guidance) ## Using pipeliine automatically uses flash attention for torch2.0 above
256
  #output_wave = tango.generate(prompt, steps, guidance)