apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

make controlnet params match others

This commit is contained in:
Sean Sube 2023-12-18 22:14:33 -06:00
parent de4b6882f1
commit 6708efa8ce
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/diffusers/pipelines/controlnet.py
View File

@ -18,7 +18,7 @@ from diffusers.pipelines.pipeline_utils import DiffusionPipeline
from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
from diffusers.schedulers import DDIMScheduler, LMSDiscreteScheduler, PNDMScheduler
from diffusers.utils import PIL_INTERPOLATION, deprecate, logging
from transformers import CLIPTokenizer
from transformers import CLIPImageProcessor, CLIPTokenizer
logger = logging.get_logger(__name__)
@ -41,6 +41,8 @@ class OnnxStableDiffusionControlNetPipeline(DiffusionPipeline):
unet: OnnxRuntimeModel,
controlnet: OnnxRuntimeModel,
scheduler: Union[DDIMScheduler, PNDMScheduler, LMSDiscreteScheduler],
safety_checker: OnnxRuntimeModel,
feature_extractor: CLIPImageProcessor,
requires_safety_checker: bool = False,
):
super().__init__()