Update forge_ipadapter.py

This commit is contained in:
lllyasviel
2024-01-31 11:25:14 -08:00
parent 1cce6cfd5b
commit 681701a924

View File

@@ -30,7 +30,7 @@ class PreprocessorClipVisionForIPAdapter(PreprocessorClipVision):
return cond
class PreprocessorInsightFaceForIPAdapter(PreprocessorClipVisionForIPAdapter):
class PreprocessorClipVisionWithInsightFaceForIPAdapter(PreprocessorClipVisionForIPAdapter):
def __init__(self, name, url, filename):
super().__init__(name, url, filename)
self.cached_insightface = None
@@ -66,7 +66,7 @@ add_supported_preprocessor(PreprocessorClipVisionForIPAdapter(
filename='CLIP-ViT-bigG.safetensors'
))
add_supported_preprocessor(PreprocessorInsightFaceForIPAdapter(
add_supported_preprocessor(PreprocessorClipVisionWithInsightFaceForIPAdapter(
name='InsightFace+CLIP-H (IPAdapter)',
url='https://huggingface.co/h94/IP-Adapter/resolve/main/models/image_encoder/model.safetensors',
filename='CLIP-ViT-H-14.safetensors'