docs: add dockerfile example

pull/1/head
Bryce 2 years ago
parent fa3673ef56
commit a1675fa599

@ -0,0 +1,10 @@
FROM python:3.10.6-slim
RUN apt-get update && apt-get install -y libgl1 libglib2.0-0
ENV PIP_DISABLE_PIP_VERSION_CHECK=1 \
PIP_ROOT_USER_ACTION=ignore
RUN pip install imaginairy
#RUN #imagine pizza

@ -108,6 +108,14 @@ imagine_image_files(prompts, outdir="./my-art")
- ~10 gb space for models to download
- A decent computer with either a CUDA supported graphics card or M1 processor.
## Running in Docker
See example Dockerfile (works on machine where you can pass the gpu into the container)
```bash
docker build . -t imaginairy
# you really want to map the cache or you end up wasting a lot of time and space redownloading the model weights
docker run -it --gpus all -v $HOME/.cache/huggingface:/root/.cache/huggingface -v $HOME/.cache/torch:/root/.cache/torch -v `pwd`/outputs:/outputs imaginairy /bin/bash
```
## Improvements from CompVis
- img2img actually does # of steps you specify
- performance optimizations

Loading…
Cancel
Save