update for ascend
This commit is contained in:
17
Dockerfile_ascend
Normal file
17
Dockerfile_ascend
Normal file
@@ -0,0 +1,17 @@
|
||||
FROM quay.io/ascend/vllm-ascend:v0.10.0rc1
|
||||
|
||||
WORKDIR /workspace/
|
||||
COPY ./model_test_caltech_http_ascend.py /workspace/
|
||||
COPY ./microsoft_beit_base_patch16_224_pt22k_ft22k /model
|
||||
|
||||
|
||||
# 安装transformers 4.46.3
|
||||
RUN python3 -m pip install --no-cache-dir transformers==4.46.3
|
||||
|
||||
|
||||
RUN python3 -m pip install flask==3.1.1
|
||||
|
||||
EXPOSE 80
|
||||
|
||||
|
||||
ENTRYPOINT ["python3", "model_test_caltech_http_ascend.py"]
|
||||
Reference in New Issue
Block a user