29 lines
879 B
Python
29 lines
879 B
Python
import folder_paths
|
|
|
|
from .utils import TimestepKeyframeGroup
|
|
from .control import load_sparsectrl
|
|
|
|
|
|
# node for SparseCtrl loading
|
|
class SparseCtrlLoaderAdvanced:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"control_net_name": (folder_paths.get_filename_list("controlnet"), ),
|
|
},
|
|
"optional": {
|
|
"tk_optional": ("TIMESTEP_KEYFRAME", ),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("CONTROL_NET", )
|
|
FUNCTION = "load_controlnet"
|
|
|
|
CATEGORY = "Adv-ControlNet 🛂🅐🅒🅝/loaders"
|
|
|
|
def load_controlnet(self, control_net_name: str, tk_optional: TimestepKeyframeGroup=None):
|
|
controlnet_path = folder_paths.get_full_path("controlnet", control_net_name)
|
|
controlnet = load_sparsectrl(controlnet_path, timestep_keyframe=tk_optional)
|
|
return controlnet
|