This commit is contained in:
Qing 2023-11-15 16:52:44 +08:00
parent 3e7a553273
commit 1571018cfd
2 changed files with 2 additions and 3 deletions

View File

@ -60,6 +60,7 @@ class Kandinsky(DiffusionInpaintModel):
output_type="np", output_type="np",
callback=self.callback, callback=self.callback,
generator=generator, generator=generator,
callback_steps=1,
).images[0] ).images[0]
output = (output * 255).round().astype("uint8") output = (output * 255).round().astype("uint8")

View File

@ -17,10 +17,8 @@ from typing import Union, List, Optional, Callable, Dict, Any
import torch import torch
import PIL.Image 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.pipelines.controlnet.pipeline_controlnet import *
from diffusers.utils import replace_example_docstring from diffusers.utils import replace_example_docstring
EXAMPLE_DOC_STRING = """ EXAMPLE_DOC_STRING = """