Skip to content
代码片段 群组 项目
提交 8515d963 编辑于 作者: comfyanonymous's avatar comfyanonymous
浏览文件

It's more logical for the StyleModelApply to append it directly.

上级 7ec1dd25
No related branches found
No related tags found
无相关合并请求
......@@ -425,33 +425,20 @@ class StyleModelLoader:
class StyleModelApply:
@classmethod
def INPUT_TYPES(s):
return {"required": {"clip_vision_output": ("CLIP_VISION_OUTPUT", ),
"style_model": ("STYLE_MODEL", )
return {"required": {"conditioning": ("CONDITIONING", ),
"style_model": ("STYLE_MODEL", ),
"clip_vision_output": ("CLIP_VISION_OUTPUT", ),
}}
RETURN_TYPES = ("CONDITIONING",)
FUNCTION = "apply_stylemodel"
CATEGORY = "conditioning/style_model"
def apply_stylemodel(self, clip_vision_output, style_model):
c = style_model.get_cond(clip_vision_output)
return ([[c, {}]], )
class ConditioningAppend:
@classmethod
def INPUT_TYPES(s):
return {"required": {"conditioning_to": ("CONDITIONING", ), "conditioning_from": ("CONDITIONING", )}}
RETURN_TYPES = ("CONDITIONING",)
FUNCTION = "append"
CATEGORY = "conditioning/style_model"
def append(self, conditioning_to, conditioning_from):
def apply_stylemodel(self, clip_vision_output, style_model, conditioning):
cond = style_model.get_cond(clip_vision_output)
c = []
to_append = conditioning_from[0][0]
for t in conditioning_to:
n = [torch.cat((t[0],to_append), dim=1), t[1].copy()]
for t in conditioning:
n = [torch.cat((t[0], cond), dim=1), t[1].copy()]
c.append(n)
return (c, )
......@@ -952,8 +939,7 @@ NODE_CLASS_MAPPINGS = {
"LoraLoader": LoraLoader,
"CLIPLoader": CLIPLoader,
"CLIPVisionEncode": CLIPVisionEncode,
"StyleModelApply":StyleModelApply,
"ConditioningAppend":ConditioningAppend,
"StyleModelApply": StyleModelApply,
"ControlNetApply": ControlNetApply,
"ControlNetLoader": ControlNetLoader,
"DiffControlNetLoader": DiffControlNetLoader,
......
0% 加载中 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册