lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-04 06:59:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update forge_ipadapter.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-31 10:54:16 -08:00
parent 7055bd17dc
commit 0449490231
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
extensions-builtin/sd_forge_ipadapter/scripts/forge_ipadapter.py
View File

@@ -18,7 +18,7 @@ class PreprocessorClipVisionForIPAdapter(PreprocessorClipVision):
self.sorting_priority = 20
def __call__(self, input_image, resolution, slider_1=None, slider_2=None, slider_3=None, **kwargs):
return dict(
cond = dict(
clip_vision=self.load_clipvision(),
image=numpy_to_pytorch(input_image),
weight_type="original",
@@ -27,6 +27,7 @@ class PreprocessorClipVisionForIPAdapter(PreprocessorClipVision):
attn_mask=None,
unfold_batch=False,
)
return cond
class PreprocessorInsightFaceForIPAdapter(Preprocessor):
@@ -41,7 +42,7 @@ class PreprocessorInsightFaceForIPAdapter(Preprocessor):
self.sorting_priority = 1
def __call__(self, input_image, resolution, slider_1=None, slider_2=None, slider_3=None, **kwargs):
return dict(
cond = dict(
insightface=opInsightFaceLoader("CPU")[0],
image=numpy_to_pytorch(input_image),
weight_type="original",
@@ -50,6 +51,7 @@ class PreprocessorInsightFaceForIPAdapter(Preprocessor):
attn_mask=None,
unfold_batch=False,
)
return cond
add_supported_preprocessor(PreprocessorClipVisionForIPAdapter(