19 lines
864 B
Plaintext
19 lines
864 B
Plaintext
FROM git.modelhub.org.cn:9443/enginex-kunlunxin/diffuser:r200-8f
|
|
|
|
WORKDIR /workspace/
|
|
COPY ./model_test_caltech_http_kunlunxin.py /workspace/
|
|
COPY ./microsoft_beit_base_patch16_224_pt22k_ft22k /model
|
|
|
|
|
|
# 安装transformers 4.46.3
|
|
RUN /root/miniconda/envs/python38_torch201_cuda/bin/python3 -m pip install --upgrade pip
|
|
RUN /root/miniconda/envs/python38_torch201_cuda/bin/python3 -m pip install --no-cache-dir -i https://pypi.tuna.tsinghua.edu.cn/simple transformers==4.46.3
|
|
RUN /root/miniconda/envs/python38_torch201_cuda/bin/python3 -m pip install flask
|
|
|
|
EXPOSE 80
|
|
|
|
ENTRYPOINT ["/root/miniconda/envs/python38_torch201_cuda/bin/python3", "model_test_caltech_http_kunlunxin.py"]
|
|
|
|
|
|
|