21 lines
690 B
Docker
21 lines
690 B
Docker
FROM git.modelhub.org.cn:9443/enginex-cambricon/mlu370-pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310
|
|
|
|
WORKDIR /app
|
|
|
|
ENV PATH=/torch/venv3/pytorch_infer/bin:/workspace/ffmpeg-mlu-v4.2.0/install/bin:/usr/local/neuware/bin:/usr/local/openmpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
|
|
|
|
RUN sed -i 's|source /torch/venv3/pytorch/bin/activate|source /torch/venv3/pytorch_infer/bin/activate|' /root/.bashrc
|
|
|
|
|
|
RUN pip install qwen-omni-utils
|
|
|
|
COPY app.py .
|
|
|
|
COPY transformers /app/transformers
|
|
RUN cd transformers && pip install -e .
|
|
|
|
ENTRYPOINT []
|
|
|
|
CMD ["/torch/venv3/pytorch_infer/bin/python", "-m", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "80"]
|
|
|