mirror of
https://github.com/deepseek-ai/Janus.git
synced 2025-04-19 10:09:00 -04:00
19 lines
451 B
Docker
19 lines
451 B
Docker
FROM python:3.10-slim
|
|
|
|
COPY . /Janus
|
|
|
|
WORKDIR /Janus
|
|
|
|
RUN pip install --no-cache-dir -r requirements.txt
|
|
RUN pip install -e .
|
|
RUN pip install --upgrade torch
|
|
|
|
ENV GRADIO_SERVER_NAME="0.0.0.0"
|
|
ENV MODEL_PATH_APP="models/models-deepseek-ai-Janus-Pro-1B"
|
|
ENV MODEL_PATH_APP_JANUSPRO="models/models-deepseek-ai-Janus-Pro-7B"
|
|
ENV HF_HOME=/Janus/models
|
|
ENV TRANSFORMERS_CACHE=/Janus/models
|
|
ENV HF_DATASETS_CACHE=/Janus/models
|
|
|
|
CMD ["python", "demo/app.py"]
|