feat(paddleocr-vl): add PaddleOCR-VL GPU Dockerfile, pin vllm, update CPU image deps, and improve entrypoint and tests
This commit is contained in:
@@ -28,7 +28,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||
# Install Python dependencies
|
||||
RUN pip install --no-cache-dir --upgrade pip && \
|
||||
pip install --no-cache-dir \
|
||||
torch==2.5.1 --index-url https://download.pytorch.org/whl/cpu && \
|
||||
torch==2.5.1 torchvision==0.20.1 --index-url https://download.pytorch.org/whl/cpu && \
|
||||
pip install --no-cache-dir \
|
||||
transformers \
|
||||
accelerate \
|
||||
@@ -37,7 +37,10 @@ RUN pip install --no-cache-dir --upgrade pip && \
|
||||
fastapi \
|
||||
uvicorn[standard] \
|
||||
python-multipart \
|
||||
httpx
|
||||
httpx \
|
||||
protobuf \
|
||||
sentencepiece \
|
||||
einops
|
||||
|
||||
# Copy server files
|
||||
COPY image_support_files/paddleocr_vl_server.py /app/paddleocr_vl_server.py
|
||||
|
||||
Reference in New Issue
Block a user