From b79a5e64a7c920395d212ba4c89c8a0e4534c29c Mon Sep 17 00:00:00 2001
From: Thomas  MENARD <thomas.menard@pasteur.fr>
Date: Fri, 3 Nov 2023 17:10:13 +0100
Subject: [PATCH]

---
 Dockerfile.cuda11 | 2 +-
 requirements.txt  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile.cuda11 b/Dockerfile.cuda11
index db2090a..c0e2f1a 100644
--- a/Dockerfile.cuda11
+++ b/Dockerfile.cuda11
@@ -7,7 +7,7 @@ WORKDIR /app
 COPY requirements.txt requirements.txt
 RUN pip3 install -r requirements.txt
 # https://github.com/marella/ctransformers#cuda
-RUN pip3 install ctransformers[cuda11]
+RUN pip3 install ctransformers[cuda]
 COPY . .
 EXPOSE 8000
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
diff --git a/requirements.txt b/requirements.txt
index be7fa66..c2ffd8d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,7 +3,7 @@ blake3==0.3.3
 certifi==2023.7.22
 charset-normalizer==3.1.0
 click==8.1.3
-ctransformers==0.2.27
+ctransformers==0.2.20
 fastapi==0.95.2
 filelock==3.12.0
 fsspec==2023.5.0
-- 
GitLab