kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

update

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lama_cleaner/model/kandinsky.py
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")

4
lama_cleaner/model/pipeline/pipeline_stable_diffusion_controlnet_inpaint.py
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 = """