hungchiayu commited on
Commit
5c8506d
·
1 Parent(s): c62fe9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ tangoflux = TangoFluxInference(name="declare-lab/TangoFlux")
26
  def gradio_generate(prompt, output_format, steps, guidance,duration=10):
27
 
28
  output_wave = tangoflux.generate(prompt,steps=steps,guidance_scale=guidance,duration=duration)
29
- output_wave = pipe(prompt,steps,guidance) ## Using pipeliine automatically uses flash attention for torch2.0 above
30
  #output_wave = tango.generate(prompt, steps, guidance)
31
  # output_filename = f"{prompt.replace(' ', '_')}_{steps}_{guidance}"[:250] + ".wav"
32
  output_wave = output_wave.audios[0]
 
26
  def gradio_generate(prompt, output_format, steps, guidance,duration=10):
27
 
28
  output_wave = tangoflux.generate(prompt,steps=steps,guidance_scale=guidance,duration=duration)
29
+ #output_wave = pipe(prompt,steps,guidance) ## Using pipeliine automatically uses flash attention for torch2.0 above
30
  #output_wave = tango.generate(prompt, steps, guidance)
31
  # output_filename = f"{prompt.replace(' ', '_')}_{steps}_{guidance}"[:250] + ".wav"
32
  output_wave = output_wave.audios[0]