9 lines
437 B
Docker
9 lines
437 B
Docker
# 注意：以下基础镜像需要从寒武纪厂商获取(联系寒武纪技术支持)
|
|
FROM <combricon_docker_image>/mic-llm-x/combricon-mlu370x4-base:v0.2.0-tgiv1.4.3-btv0.6.0-pt2.1-x86_64-ubuntu22.04-py310
|
|
ENV HF_ENDPOINT=https://hf-mirror.com
|
|
RUN pip install transformers==4.50.0 uvicorn\[standard\] fastapi
|
|
WORKDIR /app
|
|
COPY server.py /app/server.py
|
|
EXPOSE 8000
|
|
CMD ["uvicorn", "server:app", "--host", "0.0.0.0", "--port", "8000"]
|