diff --git a/Dockerfile.cuda11 b/Dockerfile.cuda11 index c0e2f1aec9766899106016269e1a7228621e84de..db2090a53617d26ec067569a118a85080971d8b1 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 a9ac00bf4e43f8d7e2df7e437cc58a076efb254d..be7fa6608466f189dacc4a1db7908113d9373e05 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