SebastianBodza commited on
Commit
3bdf08e
·
verified ·
1 Parent(s): 153080b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -134,8 +134,8 @@ def extract_speech_ids(speech_tokens_str):
134
  print(f"Unexpected token: {token_str}")
135
  return speech_ids
136
 
137
- @torch.inference_mode()
138
  @spaces.GPU(duration=30)
 
139
  def infer_with_speaker(
140
  display_name,
141
  target_text,
@@ -162,8 +162,8 @@ def infer_with_speaker(
162
  progress,
163
  )
164
 
165
- @torch.inference_mode()
166
  @spaces.GPU(duration=30)
 
167
  def gradio_infer(*args, **kwargs):
168
  return infer(*args, **kwargs)
169
 
 
134
  print(f"Unexpected token: {token_str}")
135
  return speech_ids
136
 
 
137
  @spaces.GPU(duration=30)
138
+ @torch.inference_mode()
139
  def infer_with_speaker(
140
  display_name,
141
  target_text,
 
162
  progress,
163
  )
164
 
 
165
  @spaces.GPU(duration=30)
166
+ @torch.inference_mode()
167
  def gradio_infer(*args, **kwargs):
168
  return infer(*args, **kwargs)
169