fix(api): various controlnet fixes
This commit is contained in:
parent
a716f6db1f
commit
ba9982af51
|
@ -546,6 +546,14 @@ def main(args=None) -> int:
|
|||
server.opset = args.opset
|
||||
server.token = args.token
|
||||
|
||||
# debug options
|
||||
if server.debug:
|
||||
import debugpy
|
||||
|
||||
debugpy.listen(5678)
|
||||
logger.warning("waiting for debugger")
|
||||
debugpy.wait_for_client()
|
||||
|
||||
register_plugins(server)
|
||||
|
||||
logger.info(
|
||||
|
|
|
@ -507,7 +507,7 @@ def convert_diffusion_diffusers(
|
|||
run_gc()
|
||||
|
||||
if conversion.control and not single_vae and not conversion.share_unet:
|
||||
cnet_source = torch_source or source
|
||||
cnet_source = torch_source or cache_path
|
||||
logger.info("loading and converting CNet from %s", cnet_source)
|
||||
cnet_path = convert_diffusion_diffusers_cnet(
|
||||
conversion,
|
||||
|
|
|
@ -235,7 +235,8 @@ def load_pipeline(
|
|||
pipeline_class.__name__,
|
||||
)
|
||||
pipe = pipeline_class(
|
||||
components["vae"],
|
||||
components["vae_encoder"],
|
||||
components["vae_decoder"],
|
||||
components["text_encoder"],
|
||||
components["tokenizer"],
|
||||
components["unet"],
|
||||
|
|
|
@ -160,7 +160,7 @@ def source_filter_hed(server: ServerContext, source: Image.Image) -> Image.Image
|
|||
logger.debug("running HED detection on source image")
|
||||
|
||||
hed = HEDdetector.from_pretrained(
|
||||
"lllyasviel/ControlNet",
|
||||
"lllyasviel/Annotators",
|
||||
cache_dir=server.cache_path,
|
||||
)
|
||||
image = hed(source)
|
||||
|
@ -172,7 +172,7 @@ def source_filter_scribble(server: ServerContext, source: Image.Image) -> Image.
|
|||
logger.debug("running scribble detection on source image")
|
||||
|
||||
hed = HEDdetector.from_pretrained(
|
||||
"lllyasviel/ControlNet",
|
||||
"lllyasviel/Annotators",
|
||||
cache_dir=server.cache_path,
|
||||
)
|
||||
image = hed(source, scribble=True)
|
||||
|
|
Loading…
Reference in New Issue