fix: remove flash attention

winstxnhdw · Jun 18, 2024 · 61e4a5d · 61e4a5d
1 parent 16cc58d
commit 61e4a5d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/Dockerfile.build b/Dockerfile.build
@@ -8,9 +8,8 @@ WORKDIR /
 
 COPY pyproject.toml poetry.lock ./
 
-RUN apt update
-RUN apt install -y curl
-RUN curl -sSL https://install.python-poetry.org | python -
+RUN python -m venv $POETRY_HOME
+RUN $POETRY_HOME/bin/pip install poetry
 RUN poetry install --without dev --no-root
 
 

diff --git a/capgen/transcriber/transcriber.py b/capgen/transcriber/transcriber.py
@@ -47,11 +47,7 @@ def __init__(
             'num_workers': number_of_workers,
         }
 
-        try:
-            self.model = WhisperModel(**model_parameters, flash_attention=True)
-
-        except Exception:  # pylint: disable=broad-except
-            self.model = WhisperModel(**model_parameters)
+        self.model = WhisperModel(**model_parameters)
 
     async def transcribe(self, file: str | BinaryIO, caption_format: str) -> str | None:
         """