nodes/banodoco-steerable-motion
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Import working

Browse Source
This commit is contained in:
IDGallagher
2024-04-15 18:03:18 +01:00
parent 8e3d0c8169
commit 574c737dc1
1 changed files with 10 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files
imports/ComfyUI_IPAdapter_plus/IPAdapterPlus.py
+10 -77
View File
@@ -597,7 +597,7 @@ class IPAdapterSimple:
return (ipadapter_execute(model.clone(), ipadapter['ipadapter']['model'], ipadapter['clipvision']['model'], **ipa_args), ) return (ipadapter_execute(model.clone(), ipadapter['ipadapter']['model'], ipadapter['clipvision']['model'], **ipa_args), )
class IPAdapterAdvanced: class IPAdapterAdvancedImport:
def __init__(self): def __init__(self):
self.unfold_batch = False self.unfold_batch = False
@@ -672,7 +672,7 @@ class IPAdapterAdvanced:
return (ipadapter_execute(model.clone(), ipadapter_model, clip_vision, **ipa_args), ) return (ipadapter_execute(model.clone(), ipadapter_model, clip_vision, **ipa_args), )
class IPAdapterBatch(IPAdapterAdvanced): class IPAdapterBatch(IPAdapterAdvancedImport):
def __init__(self): def __init__(self):
self.unfold_batch = True self.unfold_batch = True
@@ -696,7 +696,7 @@ class IPAdapterBatch(IPAdapterAdvanced):
} }
} }
class IPAdapterStyleComposition(IPAdapterAdvanced): class IPAdapterStyleComposition(IPAdapterAdvancedImport):
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return { return {
@@ -748,7 +748,7 @@ class IPAdapterStyleCompositionBatch(IPAdapterStyleComposition):
} }
} }
class IPAdapterFaceID(IPAdapterAdvanced): class IPAdapterFaceID(IPAdapterAdvancedImport):
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return { return {
@@ -778,7 +778,7 @@ class IPAAdapterFaceIDBatch(IPAdapterFaceID):
def __init__(self): def __init__(self):
self.unfold_batch = True self.unfold_batch = True
class IPAdapterTiled: class IPAdapterTiledImport:
def __init__(self): def __init__(self):
self.unfold_batch = False self.unfold_batch = False
@@ -913,7 +913,7 @@ class IPAdapterTiled:
return (model, torch.cat(tiles), torch.cat(masks), ) return (model, torch.cat(tiles), torch.cat(masks), )
class IPAdapterTiledBatch(IPAdapterTiled): class IPAdapterTiledBatch(IPAdapterTiledImport):
def __init__(self): def __init__(self):
self.unfold_batch = True self.unfold_batch = True
@@ -989,7 +989,7 @@ class IPAdapterEmbeds:
return (ipadapter_execute(model.clone(), ipadapter_model, clip_vision, **ipa_args), ) return (ipadapter_execute(model.clone(), ipadapter_model, clip_vision, **ipa_args), )
class IPAdapterMS(IPAdapterAdvanced): class IPAdapterMS(IPAdapterAdvancedImport):
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return { return {
@@ -1119,7 +1119,7 @@ class IPAdapterCombineEmbeds:
return (embeds, ) return (embeds, )
class IPAdapterNoise: class IPAdapterNoiseImport:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return { return {
@@ -1180,7 +1180,7 @@ class IPAdapterNoise:
return (noise, ) return (noise, )
class PrepImageForClipVision: class PrepImageForClipVisionImport:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": { return {"required": {
@@ -1344,71 +1344,4 @@ class IPAdapterWeights:
if len(weights) == 0: if len(weights) == 0:
weights = [0.0] weights = [0.0]
return (weights, ) return (weights, )
"""
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Register
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
"""
NODE_CLASS_MAPPINGS = {
# Main Apply Nodes
"IPAdapter": IPAdapterSimple,
"IPAdapterAdvanced": IPAdapterAdvanced,
"IPAdapterBatch": IPAdapterBatch,
"IPAdapterFaceID": IPAdapterFaceID,
"IPAAdapterFaceIDBatch": IPAAdapterFaceIDBatch,
"IPAdapterTiled": IPAdapterTiled,
"IPAdapterTiledBatch": IPAdapterTiledBatch,
"IPAdapterEmbeds": IPAdapterEmbeds,
"IPAdapterStyleComposition": IPAdapterStyleComposition,
"IPAdapterStyleCompositionBatch": IPAdapterStyleCompositionBatch,
"IPAdapterMS": IPAdapterMS,
# Loaders
"IPAdapterUnifiedLoader": IPAdapterUnifiedLoader,
"IPAdapterUnifiedLoaderFaceID": IPAdapterUnifiedLoaderFaceID,
"IPAdapterModelLoader": IPAdapterModelLoader,
"IPAdapterInsightFaceLoader": IPAdapterInsightFaceLoader,
"IPAdapterUnifiedLoaderCommunity": IPAdapterUnifiedLoaderCommunity,
# Helpers
"IPAdapterEncoder": IPAdapterEncoder,
"IPAdapterCombineEmbeds": IPAdapterCombineEmbeds,
"IPAdapterNoise": IPAdapterNoise,
"PrepImageForClipVision": PrepImageForClipVision,
"IPAdapterSaveEmbeds": IPAdapterSaveEmbeds,
"IPAdapterLoadEmbeds": IPAdapterLoadEmbeds,
"IPAdapterWeights": IPAdapterWeights,
}
NODE_DISPLAY_NAME_MAPPINGS = {
# Main Apply Nodes
"IPAdapter": "IPAdapter",
"IPAdapterAdvanced": "IPAdapter Advanced",
"IPAdapterBatch": "IPAdapter Batch (Adv.)",
"IPAdapterFaceID": "IPAdapter FaceID",
"IPAAdapterFaceIDBatch": "IPAdapter FaceID Batch",
"IPAdapterTiled": "IPAdapter Tiled",
"IPAdapterTiledBatch": "IPAdapter Tiled Batch",
"IPAdapterEmbeds": "IPAdapter Embeds",
"IPAdapterStyleComposition": "IPAdapter Style & Composition SDXL",
"IPAdapterStyleCompositionBatch": "IPAdapter Style & Composition Batch SDXL",
"IPAdapterMS": "IPAdapter Mad Scientist",
# Loaders
"IPAdapterUnifiedLoader": "IPAdapter Unified Loader",
"IPAdapterUnifiedLoaderFaceID": "IPAdapter Unified Loader FaceID",
"IPAdapterModelLoader": "IPAdapter Model Loader",
"IPAdapterInsightFaceLoader": "IPAdapter InsightFace Loader",
"IPAdapterUnifiedLoaderCommunity": "IPAdapter Unified Loader Community",
# Helpers
"IPAdapterEncoder": "IPAdapter Encoder",
"IPAdapterCombineEmbeds": "IPAdapter Combine Embeds",
"IPAdapterNoise": "IPAdapter Noise",
"PrepImageForClipVision": "Prep Image For ClipVision",
"IPAdapterSaveEmbeds": "IPAdapter Save Embeds",
"IPAdapterLoadEmbeds": "IPAdapter Load Embeds",
"IPAdapterWeights": "IPAdapter Weights",
}