Art generation with VQGAN + CLIP in Docker, with a simple web UI for anyone with a GPU. A simplified and expanded take on Kevin Costa's work.
aiai-artclipdockerdocker-composegpuhandcodedimagenetpythonpytorchtorchtorchvisionvqganvqgan-clip
1FROM pytorch/pytorch:latest
2
3RUN apt-get update && apt-get install -y --no-install-recommends \
4 git && \
5 rm -rf /var/lib/apt/lists/*
6
7WORKDIR /app
8
9COPY Pipfile Pipfile.lock .
10RUN pip install pipenv && \
11 pipenv install --system
12
13COPY scripts scripts
14COPY core core
15
16VOLUME /data
17
18ENV FLASK_APP=scripts/web.py
19
20CMD ["flask", "run", "--port", "3000", "--host", "0.0.0.0"]