Spaces:
Runtime error
Runtime error
liuyang
commited on
Commit
·
02f099e
1
Parent(s):
fc7179b
cuda on init
Browse files
app.py
CHANGED
|
@@ -369,7 +369,7 @@ for model_name in MODELS.keys():
|
|
| 369 |
try:
|
| 370 |
print(f"Loading WhisperX model '{model_name}'...")
|
| 371 |
whisperx_model_name = MODELS[model_name]["whisperx_name"]
|
| 372 |
-
device = "
|
| 373 |
compute_type = "float16"
|
| 374 |
|
| 375 |
model = whisperx.load_model(
|
|
@@ -390,7 +390,7 @@ print("Preloading all WhisperX alignment models...")
|
|
| 390 |
for lang in ALIGN_LANGUAGES:
|
| 391 |
try:
|
| 392 |
print(f"Loading alignment model for language '{lang}'...")
|
| 393 |
-
device = "
|
| 394 |
|
| 395 |
align_model, align_metadata = whisperx.load_align_model(
|
| 396 |
language_code=lang,
|
|
|
|
| 369 |
try:
|
| 370 |
print(f"Loading WhisperX model '{model_name}'...")
|
| 371 |
whisperx_model_name = MODELS[model_name]["whisperx_name"]
|
| 372 |
+
device = "cuda" # Load on CPU initially, will move to GPU when needed
|
| 373 |
compute_type = "float16"
|
| 374 |
|
| 375 |
model = whisperx.load_model(
|
|
|
|
| 390 |
for lang in ALIGN_LANGUAGES:
|
| 391 |
try:
|
| 392 |
print(f"Loading alignment model for language '{lang}'...")
|
| 393 |
+
device = "cuda" # Load on CPU initially, will move to GPU when needed
|
| 394 |
|
| 395 |
align_model, align_metadata = whisperx.load_align_model(
|
| 396 |
language_code=lang,
|