Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
|