Spaces:
Runtime error
Runtime error
liuyang
commited on
Commit
·
91dc7f1
1
Parent(s):
aa22724
remove cuda 0
Browse files
app.py
CHANGED
|
@@ -34,7 +34,7 @@ pipe = pipeline(
|
|
| 34 |
"automatic-speech-recognition",
|
| 35 |
model="openai/whisper-large-v3-turbo",
|
| 36 |
torch_dtype=torch.float16,
|
| 37 |
-
device="cuda
|
| 38 |
model_kwargs={"attn_implementation": "flash_attention_2"},
|
| 39 |
return_timestamps=True,
|
| 40 |
)
|
|
|
|
| 34 |
"automatic-speech-recognition",
|
| 35 |
model="openai/whisper-large-v3-turbo",
|
| 36 |
torch_dtype=torch.float16,
|
| 37 |
+
device="cuda",
|
| 38 |
model_kwargs={"attn_implementation": "flash_attention_2"},
|
| 39 |
return_timestamps=True,
|
| 40 |
)
|