liuyang commited on
Commit
02f099e
·
1 Parent(s): fc7179b

cuda on init

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = "cpu" # Load on CPU initially, will move to GPU when needed
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 = "cpu" # Load on CPU initially, will move to GPU when needed
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,