jhj0517 commited on
Commit
6532cf8
2 Parent(s): acfa385 8926685

Merge pull request #167 from jhj0517/feature/enable-finetuning-models

Browse files
modules/faster_whisper_inference.py CHANGED
@@ -24,6 +24,7 @@ class FasterWhisperInference(BaseInterface):
24
  def __init__(self):
25
  super().__init__()
26
  self.model_dir = os.path.join("models", "Whisper", "faster-whisper")
 
27
  self.current_model_size = None
28
  self.model = None
29
  self.model_paths = self.get_model_paths()
 
24
  def __init__(self):
25
  super().__init__()
26
  self.model_dir = os.path.join("models", "Whisper", "faster-whisper")
27
+ os.makedirs(self.model_dir, exist_ok=True)
28
  self.current_model_size = None
29
  self.model = None
30
  self.model_paths = self.get_model_paths()