Skip to content

Commit

Permalink
fix(docs): describe how to install inpainting model
Browse files Browse the repository at this point in the history
  • Loading branch information
ssube committed Jan 9, 2023
1 parent 1e2321f commit 2332c44
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,9 @@ This is still an early project and the instructions are a little rough, but it w
- txt2img mode
- outputs are saved to file
- show image history
- img2img mode
- mask painting
- source to mask conversion tools

## Contents

Expand Down Expand Up @@ -249,14 +252,28 @@ Download the conversion script from the `huggingface/diffusers` repository to th
Run the conversion script with your desired model(s):

```shell
# on linux:
> python convert_stable_diffusion_checkpoint_to_onnx.py --model_path="runwayml/stable-diffusion-v1-5" --output_path="./models/stable-diffusion-onnx-v1-5"

# on windows:
> python convert_stable_diffusion_checkpoint_to_onnx.py --model_path="runwayml/stable-diffusion-v1-5" --output_path=".\models\stable-diffusion-onnx-v1-5"
```

This will take a little while to convert each model. Stable diffusion v1.4 is about 6GB, v1.5 is at least 10GB or so.

You should verify that all of the steps up to this point have worked correctly by attempting to run the
`api/test-setup.py` script, which is a slight variation on the original txt2img script.

If you want to use inpainting, you will need a second model trained for that purpose:

```shell
# on linux:
> python convert_stable_diffusion_checkpoint_to_onnx.py --model_path="runwayml/stable-diffusion-inpainting" --output_path="./models/stable-diffusion-inpainting"

# on windows:
> python convert_stable_diffusion_checkpoint_to_onnx.py --model_path="runwayml/stable-diffusion-inpainting" --output_path=".\models\stable-diffusion-inpainting"
```

## Usage

### Configuring and running the server
Expand Down

0 comments on commit 2332c44

Please sign in to comment.