jhj0517 commited on
Commit
988860f
·
1 Parent(s): e862b08

Fix factory function

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_factory.py +2 -2
modules/whisper/whisper_factory.py CHANGED
@@ -59,14 +59,14 @@ class WhisperFactory:
59
  diarization_model_dir=diarization_model_dir,
60
  uvr_model_dir=uvr_model_dir
61
  )
62
- elif whisper_type in WhisperImpl.WHISPER:
63
  return WhisperInference(
64
  model_dir=whisper_model_dir,
65
  output_dir=output_dir,
66
  diarization_model_dir=diarization_model_dir,
67
  uvr_model_dir=uvr_model_dir
68
  )
69
- elif whisper_type in WhisperImpl.INSANELY_FAST_WHISPER:
70
  return InsanelyFastWhisperInference(
71
  model_dir=insanely_fast_whisper_model_dir,
72
  output_dir=output_dir,
 
59
  diarization_model_dir=diarization_model_dir,
60
  uvr_model_dir=uvr_model_dir
61
  )
62
+ elif whisper_type == WhisperImpl.WHISPER:
63
  return WhisperInference(
64
  model_dir=whisper_model_dir,
65
  output_dir=output_dir,
66
  diarization_model_dir=diarization_model_dir,
67
  uvr_model_dir=uvr_model_dir
68
  )
69
+ elif whisper_type == WhisperImpl.INSANELY_FAST_WHISPER:
70
  return InsanelyFastWhisperInference(
71
  model_dir=insanely_fast_whisper_model_dir,
72
  output_dir=output_dir,