From 99c5e3a5ae3604849d301afc387edd8fd2641e9b Mon Sep 17 00:00:00 2001 From: Eva Maxfield Brown Date: Sat, 2 Dec 2023 13:49:32 -0800 Subject: [PATCH] Update deps to fix whisper transcription --- pyproject.toml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 398c635c..f35db971 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -57,6 +57,7 @@ Documentation = "https://CouncilDataProject.github.io/cdp-backend" # https://peps.python.org/pep-0621/#dependencies-optional-dependencies [project.optional-dependencies] pipeline = [ + "curated-transformers>=0.1.1", "dask[distributed]>=2021.7.0", "faster-whisper==0.3.0", "ffmpeg-python>=0.2.0", @@ -75,7 +76,7 @@ pipeline = [ "spacy>=3.4", "spacy-transformers>=1.1", "tika==2.6.0", - "torch>=1.10", + "torch>=1.10,<2", "tqdm>=4.62", "transformers>=4.16", "vimeo_downloader>=0.4",