Feat: add kotoba-whisper, etc

2024-05-26 22:03:19 +09:00
parent e27bd4dfbc
commit e5c189125f
7 changed files with 101 additions and 36 deletions
--- a/transcribe.py
+++ b/transcribe.py
@@ -76,6 +76,7 @@ def transcribe_files_with_hf_whisper(
        batch_size=batch_size,
        torch_dtype=torch.float16,
        device="cuda",
+        trust_remote_code=True,
        # generate_kwargs=generate_kwargs,
    )
    if initial_prompt is not None: