31 lines
769 B
Docker
31 lines
769 B
Docker
|
FROM python:3.12-slim
|
||
|
|
|
||
|
|
WORKDIR /home/ooin/st/app
|
||
|
|
|
||
|
|
RUN apt-get update && \
|
||
|
|
apt-get install -y --no-install-recommends \
|
||
|
|
build-essential \
|
||
|
|
libssl-dev \
|
||
|
|
libffi-dev \
|
||
|
|
libxml2-dev \
|
||
|
|
libxslt1-dev \
|
||
|
|
zlib1g-dev \
|
||
|
|
libjpeg-dev \
|
||
|
|
libopenblas-dev \
|
||
|
|
libopenmpi-dev \
|
||
|
|
&& rm -rf /var/lib/apt/lists/*
|
||
|
|
|
||
|
|
COPY requirements.txt .
|
||
|
|
|
||
|
|
RUN pip install --upgrade pip
|
||
|
|
|
||
|
|
RUN pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 flash-attn==2.7.2.post1 -f https://download.pytorch.org/whl/torch_stable.html
|
||
|
|
|
||
|
|
RUN pip install --no-cache-dir -r requirements.txt \
|
||
|
|
--index-url https://pypi.tuna.tsinghua.edu.cn/simple/ --timeout 100 --retries 5
|
||
|
|
|
||
|
|
COPY . .
|
||
|
|
|
||
|
|
EXPOSE 80
|
||
|
|
|
||
|
|
CMD ["uvicorn", "pipeline_UI3+audio:app", "--host", "0.0.0.0", "--port", "80"]
|