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