/stablediffusion-infinity

Outpainting with Stable Diffusion on an infinite canvas

Primary LanguageJupyter Notebook

stablediffusion-infinity

Open In Colab

Start with init_image:

Girl.with.a.Pearl.Earring.mp4

Start with text2img:

demo.mp4

Outpainting with Stable Diffusion on an infinite canvas.

It is recommended run the notebook on a local server for better interactive control.

The notebook might work on Windows (untested) and Apple Silicon devices (untested, check guide here: https://huggingface.co/docs/diffusers/optimization/mps).

Setup environment

setup with environment.yml

git clone --recurse-submodules https://github.com/lkwq007/stablediffusion-infinity
cd stablediffusion-infinity
conda env create -f environment.yml

if the environment.yml doesn't work for you, you may install dependencies manually:

conda create -n sd-inf python=3.10
conda activate sd-inf
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
conda install scipy
conda install -c conda-forge jupyterlab 
conda install -c conda-forge ipycanvas
conda install -c conda-forge diffusers transformers ftfy
pip install opencv-python

Note that opencv library is required for PyPatchMatch. You may need to install opencv by yourself. If no opencv installed, the patch_match option (usually better quality) won't work.

How-to

conda activate sd-inf
huggingface-cli login # ignore this if you have already logged in
jupyter lab
# and then open stablediffusion_infinity.ipynb and run cells