Spaces:
Sleeping
Sleeping
gpt-omni
commited on
Commit
•
2a8e1b5
1
Parent(s):
f8c4838
update
Browse files
app.py
CHANGED
|
@@ -194,7 +194,7 @@ def run_AT_batch_stream(
|
|
| 194 |
mel, leng = load_audio(audio_path)
|
| 195 |
audio_feature, input_ids = get_input_ids_whisper_ATBatch(mel, leng, whispermodel, device)
|
| 196 |
T = input_ids[0].size(1)
|
| 197 |
-
device = input_ids[0].device
|
| 198 |
|
| 199 |
assert max_returned_tokens > T, f"max_returned_tokens {max_returned_tokens} should be greater than audio length {T}"
|
| 200 |
|
|
|
|
| 194 |
mel, leng = load_audio(audio_path)
|
| 195 |
audio_feature, input_ids = get_input_ids_whisper_ATBatch(mel, leng, whispermodel, device)
|
| 196 |
T = input_ids[0].size(1)
|
| 197 |
+
# device = input_ids[0].device
|
| 198 |
|
| 199 |
assert max_returned_tokens > T, f"max_returned_tokens {max_returned_tokens} should be greater than audio length {T}"
|
| 200 |
|