Mike Howles
b679a5d6a2
|
3 years ago | |
---|---|---|
.. | ||
main | 3 years ago | |
prep | 3 years ago | |
README.md | 3 years ago |
README.md
Docker
Introduction
This is a Docker build file that will preinstall dependencies, packages, Git repos, and pre-cache the large model files needed by Disco Diffusion.
TO-DO:
- Make container actually accept parameters on run. Right now you'll just be seeing lighthouses.
Change Log
-
1.0
Initial build file created based on the DD 5.1 Git repo. This initial build is deliberately meant to work touch-free of any of the existing Python code written. It does handle some of the pre-setup tasks already done in the Python code such as pip packages, Git clones, and even pre-caching the model files for faster launch speed.
Build the Prep Image
The prep image is broken out from the main
folder's Dockerfile
to help with long build context times (or wget download times after intitial build.) This prep image build contains all the large model files required by Disco Diffusion.
From a terminal in the docker/prep
directory, run:
docker build -t disco-diffusion-prep:5.1 .
From a terminal in the docker/main
directory, run:
Build the Image
From a terminal, run:
docker build -t disco-diffusion:5.1 .
Run as a Container
This example runs Disco Diffusion in a Docker container. It maps images_out
and init_images
to the container's working directory to access by the host OS.
docker run --rm -it \
-v $(echo ~)/disco-diffusion/images_out:/workspace/code/images_out \
-v $(echo ~)/disco-diffusion/init_images:/workspace/code/init_images \
--gpus=all \
--name="disco-diffusion" --ipc=host \
--user $(id -u):$(id -g) \
disco-diffusion:5.1 python disco-diffusion/disco.py
Passing Parameters
This will be added after conferring with repo authors.