remove CUDA
This commit is contained in:
@@ -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"]
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user