From 2332c44cee05e1b98821041c6ca343cef606bea2 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 8 Jan 2023 21:14:47 -0600 Subject: [PATCH] fix(docs): describe how to install inpainting model --- README.md | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/README.md b/README.md index 154f06d9..37fc621f 100644 --- a/README.md +++ b/README.md @@ -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 @@ -249,7 +252,11 @@ 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. @@ -257,6 +264,16 @@ This will take a little while to convert each model. Stable diffusion v1.4 is ab 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