diff --git a/scripts/faceid.py b/scripts/faceid.py index 9e609a527..e2489226c 100644 --- a/scripts/faceid.py +++ b/scripts/faceid.py @@ -52,6 +52,10 @@ class Script(scripts.Script): onnxruntime.set_default_logger_severity(3) app.prepare(ctx_id=0, det_thresh=0.5, det_size=(640, 640)) + if isinstance(image, str): + from modules.api.api import decode_base64_to_image + image = decode_base64_to_image(image) + image = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR) faces = app.get(image) if len(faces) == 0: diff --git a/scripts/ipadapter.py b/scripts/ipadapter.py index ab6b89f08..4972a6aaf 100644 --- a/scripts/ipadapter.py +++ b/scripts/ipadapter.py @@ -115,5 +115,10 @@ class Script(scripts.Script): else: shared.log.debug(f'IP adapter cache: adapter="{adapter}" scale={scale} image={image}') shared.sd_model.set_ip_adapter_scale(scale) + + if isinstance(image, str): + from modules.api.api import decode_base64_to_image + image = decode_base64_to_image(image) + p.task_args['ip_adapter_image'] = p.batch_size * [image] p.extra_generation_params["IP Adapter"] = f'{adapter}:{scale}'