Home

Development Guide

Requirements

• Linux
• Python 3.8.16+
• GPU Support

Local setup

1. Clone the repository:

git clone https://github.com/h2oai/wave-image-styler.git

2. Download the models, install the requirements, and set up the environment with:

• make setup
  

• Diffusion model registration:

  • Create an account on Hugging Face (Helps keep track of the safe use of the model)
  • Head over to https://huggingface.co/runwayml/stable-diffusion-v1-5 and accept the licensing agreement.
  • Install git-lfs
  • cd models
  • git clone https://huggingface.co/runwayml/stable-diffusion-v1-5
  • cd stable-diffusion-v1-5
  • git lfs pull

• DALL-E mini model registration:

  • cd models
  • git clone https://huggingface.co/dalle-mini/dalle-mini
  • cd dalle-mini
  • git lfs pull
  • cd ..
  • git clone https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384
  • cd vqgan_imagenet_f16_16384
  • git lfs pull

• ControlNet models:

  • cd models
  • wget https://huggingface.co/lllyasviel/ControlNet/resolve/main/models/control_sd15_canny.pth
  • wget https://huggingface.co/lllyasviel/ControlNet/resolve/main/models/control_sd15_scribble.pth

Running the app

To run the app with active reload (automatically reload when changes are made):

make dev

Else, if active reload is not required:

make run