medium model.

brave-experiments · Aug 14, 2024 · 08610b4 · 08610b4
1 parent 142973a
commit 08610b4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/configuration.yaml b/src/configuration.yaml
@@ -2,4 +2,4 @@ runners:
   resources:
     cpu: 4
     nvidia.com/gpu: 1
-  workers_per_resource: 10
+  workers_per_resource: 12
diff --git a/src/runners/audio_transcriber.py b/src/runners/audio_transcriber.py
@@ -136,15 +136,15 @@ def __init__(self):
 
         print(self.device, " ", compute_type)
 
-        model = "base.en"
+        model = "medium"
         self.model = whisper.load_model(
             whisper_arch=model, device=self.device, compute_type=compute_type
         )
 
     def transcribe(self, audios):
         result = self.model.transcribe(
             audios,
-            batch_size=8,
+            batch_size=10,
             language="en",
             print_progress=True,
             combined_progress=True,