Debug for Whisper Extractor (#48)

whisper extractor: avoid unbound case of the download_root
open-mmlab · Dec 20, 2023 · 4542a2d · 4542a2d
1 parent d76cf2d
commit 4542a2d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/processors/content_extractor.py b/processors/content_extractor.py
@@ -198,6 +198,11 @@ def load_model(self):
             elif os.path.isdir(self.cfg.preprocess.whisper_model_path):
                 # "pretrained/whisper"
                 download_root = self.cfg.preprocess.whisper_model_path
+            else:
+                # if the path does not exist, download the model to the path
+                download_root = self.cfg.preprocess.whisper_model_path
+                if download_root.endswith(".pt"):
+                    download_root = os.path.dirname(download_root)
         else:
             download_root = None