📈 Make worker thread count configurable

bugbakery · Dec 4, 2023 · 5e56465 · 5e56465
1 parent 25a8fb2
commit 5e56465
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/worker/transcribee_worker/config.py b/worker/transcribee_worker/config.py
@@ -26,6 +26,8 @@ class Settings(BaseSettings):
 
     KEEPALIVE_INTERVAL: float = 0.5
 
+    CPU_THREADS: int = 4
+
     class Config:
         env_file = ".env"
 

diff --git a/worker/transcribee_worker/whisper_transcribe.py b/worker/transcribee_worker/whisper_transcribe.py
@@ -175,6 +175,7 @@ def transcribe_clean(
         strict_sentence_paragraphs,
     )
     model = WhisperModel(
+        cpu_threads=settings.CPU_THREADS,
         model_size_or_path=model_name,
         download_root=str((settings.MODELS_DIR / "faster_whisper").absolute()),
     )