From a94223feb6d7e04f2c4a43c6d64fbfa710743bd2 Mon Sep 17 00:00:00 2001
From: Thomas  MENARD <thomas.menard@pasteur.fr>
Date: Fri, 3 Nov 2023 17:05:18 +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 c0e2f1a..db2090a 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[cuda]
+RUN pip3 install ctransformers[cuda11]
 COPY . .
 EXPOSE 8000
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
diff --git a/requirements.txt b/requirements.txt
index a9ac00b..be7fa66 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
+ctransformers==0.2.27
 fastapi==0.95.2
 filelock==3.12.0
 fsspec==2023.5.0
-- 
GitLab