24 lines
580 B
Docker
24 lines
580 B
Docker
|
FROM rocm/pytorch:rocm6.3.2_ubuntu24.04_py3.12
|
||
|
|
|
||
|
|
WORKDIR /app
|
||
|
|
|
||
|
|
RUN apt-get update && apt-get install -y ffmpeg && rm -rf /var/lib/apt/lists/*
|
||
|
|
|
||
|
|
RUN pip install --no-cache-dir \
|
||
|
|
torch transformers accelerate \
|
||
|
|
boson_multimodal \
|
||
|
|
soundfile librosa \
|
||
|
|
pyannote.audio \
|
||
|
|
llama-cpp-python \
|
||
|
|
numpy aiofiles \
|
||
|
|
sqlalchemy aiosqlite \
|
||
|
|
pydantic pydantic-settings \
|
||
|
|
celery[redis] redis
|
||
|
|
|
||
|
|
ENV PYTHONPATH=/app
|
||
|
|
ENV HUGGINGFACE_HUB_CACHE=/data/hf_cache
|
||
|
|
|
||
|
|
COPY backend/ .
|
||
|
|
|
||
|
|
CMD ["celery", "-A", "app.tasks.celery_app", "worker", "--loglevel=info", "--concurrency=1"]
|