Pārlūkot izejas kodu

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

for more information, see https://pre-commit.ci
pre-commit-ci[bot] 2 gadi atpakaļ
vecāks
revīzija
8c30ea58c6
3 mainītis faili ar 8 papildinājumiem un 4 dzēšanām
  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
 /*.wav
 /results
 /results
 /data
 /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__":
 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")
 @click.option("--language", default="ZH", help="Language of the transcription")
 def main(model_size, audio_dir, save_dir, sample_rate, device, language):
 def main(model_size, audio_dir, save_dir, sample_rate, device, language):
     logger.info("Loading / Downloading OpenAI Whisper model...")
     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.")
     logger.info("Model loaded.")
 
 
     save_path = Path(save_dir)
     save_path = Path(save_dir)
@@ -106,4 +110,4 @@ def main(model_size, audio_dir, save_dir, sample_rate, device, language):
 
 
 
 
 if __name__ == "__main__":
 if __name__ == "__main__":
-    main()
+    main()