diff --git a/docker-compose.yml b/docker-compose.yml index af00588..e0f16ff 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -51,7 +51,7 @@ services: <<: *base_service profiles: ["lstein"] build: ./services/lstein/ - image: sd-lstein:6 + image: sd-lstein:7 environment: - PRELOAD=true - CLI_ARGS=--max_loaded_models=1 diff --git a/services/lstein/Dockerfile b/services/lstein/Dockerfile index 3746a8b..a988a61 100644 --- a/services/lstein/Dockerfile +++ b/services/lstein/Dockerfile @@ -1,37 +1,35 @@ # syntax=docker/dockerfile:1 -FROM continuumio/miniconda3:4.12.0 +FROM python:3.10-slim SHELL ["/bin/bash", "-ceuxo", "pipefail"] ENV DEBIAN_FRONTEND=noninteractive PIP_EXISTS_ACTION=w PIP_PREFER_BINARY=1 PIP_NO_CACHE_DIR=1 -# now it requires python3.9 -RUN conda install python=3.9 && conda clean -a -y -RUN conda install cudatoolkit=11.6 -c conda-forge && conda clean -a -y -RUN conda install pytorch==1.12.1 -c pytorch && conda clean -a -y -RUN apt-get update && apt install fonts-dejavu-core -y && apt-get clean +RUN pip install torch==1.13.0 torchvision --extra-index-url https://download.pytorch.org/whl/cu117 + +RUN apt-get update && apt-get install git -y && apt-get clean + +RUN git clone https://github.com/invoke-ai/InvokeAI.git /stable-diffusion + +WORKDIR /stable-diffusion RUN <