FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel

WORKDIR /app

RUN apt-get update && apt-get install -y \
    git \
    && rm -rf /var/lib/apt/lists/*

RUN git clone --recurse-submodules https://github.com/ZHZisZZ/dllm.git /tmp/dllm && \
    pip install --no-cache-dir -e /tmp/dllm && \
    pip install --no-cache-dir -e /tmp/dllm/lm-evaluation-harness

RUN pip install --no-cache-dir flask

COPY app.py .

ENV MODEL_NAME=dllm-hub/Qwen3-0.6B-diffusion-bd3lm-v0.1
ENV PORT=7860

EXPOSE 7860

CMD ["python", "app.py"]