From 78c8d8dbdbf06d0b109f9db9a4f7a9739fab9130 Mon Sep 17 00:00:00 2001 From: Qing Date: Tue, 14 Nov 2023 14:04:06 +0800 Subject: [PATCH] update --- .../pipeline/pipeline_stable_diffusion_controlnet_inpaint.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/lama_cleaner/model/pipeline/pipeline_stable_diffusion_controlnet_inpaint.py b/lama_cleaner/model/pipeline/pipeline_stable_diffusion_controlnet_inpaint.py index f5883b8..b1d3b87 100644 --- a/lama_cleaner/model/pipeline/pipeline_stable_diffusion_controlnet_inpaint.py +++ b/lama_cleaner/model/pipeline/pipeline_stable_diffusion_controlnet_inpaint.py @@ -11,14 +11,17 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. +from typing import Union, List, Optional, Callable, Dict, Any # Copy from https://github.com/mikonvergence/ControlNetInpaint/blob/main/src/pipeline_stable_diffusion_controlnet_inpaint.py import torch import PIL.Image import numpy as np +from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_controlnet import * +from diffusers.utils import replace_example_docstring EXAMPLE_DOC_STRING = """ Examples: