hungchiayu commited on
Commit
c62fe9f
·
1 Parent(s): 687c218

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ tangoflux = TangoFluxInference(name="declare-lab/TangoFlux")
25
  @spaces.GPU(duration=15)
26
  def gradio_generate(prompt, output_format, steps, guidance,duration=10):
27
 
28
- output_wave = tangoflux.generate(prompt,steps=steps,guidance=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"
 
25
  @spaces.GPU(duration=15)
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"