change controlnet depth preprocessor

This commit is contained in:
Qing
2024-01-10 21:25:51 +08:00
parent dfcb9017f1
commit a869982d13
2 changed files with 5 additions and 6 deletions

View File

@@ -93,8 +93,9 @@ class ControlNet(DiffusionInpaintModel):
model_info.path,
controlnet=controlnet,
load_safety_checker=not disable_nsfw_checker,
torch_dtype=torch_dtype,
**model_kwargs,
).to(torch_dtype)
)
else:
self.model = handle_from_pretrained_exceptions(
PipeClass.from_pretrained,