FROM runpod/pytorch:2.1.0-py3.10-cuda11.8.0-devel-ubuntu22.04
WORKDIR /app
RUN apt-get update && apt-get install -y \
git \
&& rm -rf /var/lib/apt/lists/*
COPY requirements.txt /app/
RUN pip install --no-cache-dir diffusers accelerate -r requirements.txt
COPY . /app
ENV MODEL_DEVICE=gpu
COPY --chmod=755 start.sh /start.sh
CMD ["/start.sh"]
FROM runpod/pytorch:2.1.0-py3.10-cuda11.8.0-devel-ubuntu22.04
WORKDIR /app
RUN apt-get update && apt-get install -y \
git \
&& rm -rf /var/lib/apt/lists/*
COPY requirements.txt /app/
RUN pip install --no-cache-dir diffusers accelerate -r requirements.txt
COPY . /app
ENV MODEL_DEVICE=gpu
COPY --chmod=755 start.sh /start.sh
CMD ["/start.sh"]