remove CUDA

This commit is contained in:
cesnimda
2026-03-21 22:12:47 +01:00
parent 6d9e6ca8ec
commit 91f6361055
2 changed files with 13 additions and 7 deletions
+8 -1
View File
@@ -1,7 +1,14 @@
FROM python:3.11-slim
ENV PIP_NO_CACHE_DIR=1 \
PIP_DISABLE_PIP_VERSION_CHECK=1 \
PYTHONUNBUFFERED=1 \
TRANSFORMERS_NO_TF=1 \
HF_HUB_DISABLE_TELEMETRY=1
WORKDIR /app
COPY requirements.txt ./
RUN pip install --no-cache-dir -r requirements.txt
RUN pip install --upgrade pip setuptools wheel \
&& pip install --extra-index-url https://download.pytorch.org/whl/cpu torch==2.5.1 \
&& pip install -r requirements.txt
COPY . .
EXPOSE 8001
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8001"]
+5 -6
View File
@@ -1,6 +1,5 @@
fastapi>=0.85
uvicorn[standard]>=0.18
transformers>=4.30
torch>=1.13
cachetools>=5.0
pydantic>=1.10
fastapi==0.115.12
uvicorn[standard]==0.34.0
transformers==4.48.3
cachetools==5.5.2
pydantic==2.10.6