Spaces:
Running
on
T4
Running
on
T4
gpt-omni
commited on
Commit
•
aab4898
1
Parent(s):
56fd6ef
update
Browse files
app.py
CHANGED
|
@@ -173,6 +173,7 @@ def generate_audio_data(snac_tokens, snacmodel, device=None):
|
|
| 173 |
return audio_data
|
| 174 |
|
| 175 |
|
|
|
|
| 176 |
@torch.inference_mode()
|
| 177 |
def run_AT_batch_stream(
|
| 178 |
audio_path,
|
|
|
|
| 173 |
return audio_data
|
| 174 |
|
| 175 |
|
| 176 |
+
@spaces.GPU
|
| 177 |
@torch.inference_mode()
|
| 178 |
def run_AT_batch_stream(
|
| 179 |
audio_path,
|