sd: load single file model using local config
This commit is contained in:
@@ -13,6 +13,7 @@ from .helper.controlnet_preprocess import (
|
||||
make_inpaint_control_image,
|
||||
)
|
||||
from .helper.cpu_text_encoder import CPUTextEncoderWrapper
|
||||
from .original_sd_configs import get_config_files
|
||||
from .utils import (
|
||||
get_scheduler,
|
||||
handle_from_pretrained_exceptions,
|
||||
@@ -101,6 +102,7 @@ class ControlNet(DiffusionInpaintModel):
|
||||
controlnet=controlnet,
|
||||
load_safety_checker=not disable_nsfw_checker,
|
||||
torch_dtype=torch_dtype,
|
||||
config_files=get_config_files(),
|
||||
**model_kwargs,
|
||||
)
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user