Просмотр исходного кода

[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
pre-commit-ci[bot] 2 лет назад
Родитель
Сommit
8c30ea58c6
3 измененных файлов с 8 добавлено и 4 удалено
  1. 1 1
      .gitignore
  2. 1 1
      tools/merge_asr_files.py
  3. 6 2
      tools/whisper_asr.py

+ 1 - 1
.gitignore

@@ -15,4 +15,4 @@ filelists
 /*.wav
 /results
 /data
-/.idea
+/.idea

+ 1 - 1
tools/merge_asr_files.py

@@ -52,4 +52,4 @@ def merge_and_delete_files(save_dir, original_files):
 
 
 if __name__ == "__main__":
-    merge_and_delete_files("/made/by/spicysama/laziman", [__file__])
+    merge_and_delete_files("/made/by/spicysama/laziman", [__file__])

+ 6 - 2
tools/whisper_asr.py

@@ -50,7 +50,11 @@ from fish_speech.utils.file import AUDIO_EXTENSIONS, list_files
 @click.option("--language", default="ZH", help="Language of the transcription")
 def main(model_size, audio_dir, save_dir, sample_rate, device, language):
     logger.info("Loading / Downloading OpenAI Whisper model...")
-    model = whisper.load_model(name=model_size, device=device, download_root=str(Path(".cache/whisper").resolve()))
+    model = whisper.load_model(
+        name=model_size,
+        device=device,
+        download_root=str(Path(".cache/whisper").resolve()),
+    )
     logger.info("Model loaded.")
 
     save_path = Path(save_dir)
@@ -106,4 +110,4 @@ def main(model_size, audio_dir, save_dir, sample_rate, device, language):
 
 
 if __name__ == "__main__":
-    main()
+    main()