Initial program

.gitignore

@@ -1,154 +1,2 @@
-# ---> Python
-# Byte-compiled / optimized / DLL files
-__pycache__/
-*.py[cod]
-*$py.class
-
-# C extensions
-*.so
-
-# Distribution / packaging
-.Python
-build/
-develop-eggs/
-dist/
-downloads/
-eggs/
-.eggs/
-lib/
-lib64/
-parts/
-sdist/
-var/
-wheels/
-share/python-wheels/
-*.egg-info/
-.installed.cfg
-*.egg
-MANIFEST
-
-# PyInstaller
-#  Usually these files are written by a python script from a template
-#  before PyInstaller builds the exe, so as to inject date/other infos into it.
-*.manifest
-*.spec
-
-# Installer logs
-pip-log.txt
-pip-delete-this-directory.txt
-
-# Unit test / coverage reports
-htmlcov/
-.tox/
-.nox/
-.coverage
-.coverage.*
-.cache
-nosetests.xml
-coverage.xml
-*.cover
-*.py,cover
-.hypothesis/
-.pytest_cache/
-cover/
-
-# Translations
-*.mo
-*.pot
-
-# Django stuff:
-*.log
-local_settings.py
-db.sqlite3
-db.sqlite3-journal
-
-# Flask stuff:
-instance/
-.webassets-cache
-
-# Scrapy stuff:
-.scrapy
-
-# Sphinx documentation
-docs/_build/
-
-# PyBuilder
-.pybuilder/
-target/
-
-# Jupyter Notebook
-.ipynb_checkpoints
-
-# IPython
-profile_default/
-ipython_config.py
-
-# pyenv
-#   For a library or package, you might want to ignore these files since the code is
-#   intended to run in multiple environments; otherwise, check them in:
-# .python-version
-
-# pipenv
-#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
-#   However, in case of collaboration, if having platform-specific dependencies or dependencies
-#   having no cross-platform support, pipenv may install dependencies that don't work, or not
-#   install all needed dependencies.
-#Pipfile.lock
-
-# poetry
-#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
-#   This is especially recommended for binary packages to ensure reproducibility, and is more
-#   commonly ignored for libraries.
-#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
-#poetry.lock
-
-# PEP 582; used by e.g. github.com/David-OConnor/pyflow
-__pypackages__/
-
-# Celery stuff
-celerybeat-schedule
-celerybeat.pid
-
-# SageMath parsed files
-*.sage.py
-
-# Environments
 .env
-.venv
-env/
-venv/
-ENV/
-env.bak/
-venv.bak/
-
-# Spyder project settings
-.spyderproject
-.spyproject
-
-# Rope project settings
-.ropeproject
-
-# mkdocs documentation
-/site
-
-# mypy
-.mypy_cache/
-.dmypy.json
-dmypy.json
-
-# Pyre type checker
-.pyre/
-
-# pytype static type analyzer
-.pytype/
-
-# Cython debug symbols
-cython_debug/
-
-# PyCharm
-#  JetBrains specific template is maintainted in a separate JetBrains.gitignore that can
-#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
-#  and can be added to the global gitignore or merged into this file.  For a more nuclear
-#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
-#.idea/
-
+images/

Dockerfile

@@ -0,0 +1,14 @@
+FROM nvidia/cuda:11.6.0-base-ubuntu20.04
+RUN apt-get update && apt-get install python3 python3-pip -y
+
+RUN pip3 install --upgrade diffusers transformers scipy python-dotenv cuda-python && \
+    pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117 && \
+    pip install torch==1.11.0+cu115 torchvision==0.12.0+cu115 torchaudio==0.11.0+cu115 -f https://download.pytorch.org/whl/torch_stable.html
+
+WORKDIR /app
+
+COPY main.py /app/main.py
+VOLUME /root/.cache/huggingface/diffusers/
+
+
+CMD [ "python3", "main.py" ]

README.md

@@ -1,2 +1,22 @@
-# stablediffusion
+# Stable Diffusion Example
+
+
+## Install Docker
+
+```
+curl https://get.docker.com | sh
+
+distribution=$(. /etc/os-release;echo $ID$VERSION_ID) 
+&& curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - 
+&& curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
+
+sudo apt-get update
+sudo apt-get install -y nvidia-docker2 docker-compose
+```
+
+## Run
+
+```
+sudo docker-compose up --build
+```
 

docker-compose.yaml

@@ -0,0 +1,11 @@
+version: '3.9'
+
+services:
+  app:
+    image: stablediffusion
+    build: .
+    runtime: nvidia
+    env_file:
+      - .env
+    volumes:
+      - ./images:/images

main.py

@@ -0,0 +1,30 @@
+import torch
+from diffusers import StableDiffusionPipeline
+from dotenv import load_dotenv
+from os import getenv
+from PIL.PngImagePlugin import PngInfo
+import uuid
+from random import randint
+
+load_dotenv()
+
+pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=getenv("TOKEN"))
+pipe.to("cuda")
+
+
+prompt = "red horse jumping over a dog"
+
+seed = randint(0, 2**64)
+generator = torch.Generator("cuda").manual_seed(seed)
+resp = pipe(prompt, generator=generator)
+print(resp)
+image = resp.images[0]
+
+metadata = PngInfo()
+metadata.add_text("text", prompt)
+metadata.add_text("seed", str(seed))
+
+id = str(uuid.uuid4())
+print(id)
+
+image.save(f'/images/{id}.png', pnginfo=metadata)