build: add numpy to server dependency

winstxnhdw · Jun 18, 2024 · 7e51866 · 7e51866
1 parent 61e4a5d
commit 7e51866
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/capgen/transcriber/transcriber.py b/capgen/transcriber/transcriber.py
@@ -28,7 +28,7 @@ class Transcriber:
     Methods
     -------
     transcribe(file: str | BinaryIO, caption_format: str) -> str | None:
-        converts transcription segments into a SRT file
+        converts transcription segments into a specific caption format
     """
 
     __slots__ = ('model',)
@@ -47,7 +47,11 @@ def __init__(
             'num_workers': number_of_workers,
         }
 
-        self.model = WhisperModel(**model_parameters)
+        try:
+            self.model = WhisperModel(**model_parameters, flash_attention=True)
+
+        except ValueError:
+            self.model = WhisperModel(**model_parameters)
 
     async def transcribe(self, file: str | BinaryIO, caption_format: str) -> str | None:
         """

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -36,7 +36,6 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = "^3.12"
 faster-whisper = "^1.0.2"
-numpy = "<2.0.0"
 nvidia-cudnn-cu12 = { version = "8.9.7.29", optional = true }
 nvidia-cublas-cu12 = { version = "^12.5.2.13", optional = true }
 
@@ -49,6 +48,7 @@ gunicorn = "^22.0.0"
 uvicorn = "^0.30.0"
 httptools = "^0.6.1"
 uvloop = "^0.19.0"
+numpy = "<2.0.0"
 
 [tool.poetry.group.dev.dependencies]
 pyright = "^1.1.367"