Nithya commited on
Commit
43301d8
·
1 Parent(s): a16f46b

reduced GPU time

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -108,7 +108,7 @@ def generate_audio(audio_model, f0s, invert_audio_fn, singers=[3], num_steps=100
108
 
109
  return audio
110
 
111
- @spaces.GPU(duration=180)
112
  def generate(pitch, num_samples=1, num_steps=100, singers=[3], outfolder='temp', audio_seq_len=750, pitch_qt=None ):
113
 
114
  logging.log(logging.INFO, 'Generate function')
@@ -153,7 +153,7 @@ audio_model, audio_qt, audio_seq_len, invert_audio_fn = load_audio_fns(
153
  )
154
  partial_generate = partial(generate, num_samples=1, num_steps=100, singers=[3], outfolder=None, pitch_qt=pitch_qt) # generate function with default arguments
155
 
156
- @spaces.GPU(duration=180)
157
  def set_guide_and_generate(audio):
158
  global selected_prime, pitch_task_fn
159
 
 
108
 
109
  return audio
110
 
111
+ @spaces.GPU(duration=120)
112
  def generate(pitch, num_samples=1, num_steps=100, singers=[3], outfolder='temp', audio_seq_len=750, pitch_qt=None ):
113
 
114
  logging.log(logging.INFO, 'Generate function')
 
153
  )
154
  partial_generate = partial(generate, num_samples=1, num_steps=100, singers=[3], outfolder=None, pitch_qt=pitch_qt) # generate function with default arguments
155
 
156
+ @spaces.GPU(duration=120)
157
  def set_guide_and_generate(audio):
158
  global selected_prime, pitch_task_fn
159