From 9e42899f44bbe8b2e41baed4823399e24fd2fb62 Mon Sep 17 00:00:00 2001
From: zxl <eng.lzx@foxmail.com>
Date: Tue, 28 Nov 2023 14:41:57 +0800
Subject: [PATCH] feat: support relatetive iplora pth
---
scripts/controlnet.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/scripts/controlnet.py b/scripts/controlnet.py
index 5f91668..7c69491 100644
--- a/scripts/controlnet.py
+++ b/scripts/controlnet.py
@@ -794,7 +794,9 @@ class Script(scripts.Script, metaclass=(
if control_model_type is ControlModelType.ControlNet:
global_average_pooling = model_net.control_model.global_average_pooling
-
+
+ if unit.multi_references_dir != '':
+ unit.multi_references_dir = "models/iplora/" + unit.multi_references_dir
# JIAHUI'S MODIFY
preprocessor_resolution = None
# reference mode priority is 'single' > 'Multiple'
@@ -837,9 +839,9 @@ class Script(scripts.Script, metaclass=(
clip_outputs = []
if control_model_type == ControlModelType.IPAdapter:
- logger.info("now in ipadter unit_multi_reference_dir: ", unit.multi_references_dir)
+ logger.info("now in ipadter unit_multi_reference_dir: %s", unit.multi_references_dir)
if unit.image is not None or not os.path.exists(unit.multi_references_dir):
- logger.info("unit_multi_reference_dir: ", unit.multi_references_dir)
+ logger.info("unit_multi_reference_dir: %s", unit.multi_references_dir)
for input_image, preprocessor_resolution in zip(input_images, preprocessor_resolution_list):
if isinstance(input_image, Image.Image):
input_image = np.array(input_image)
--
GitLab