xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-07-08 16:07:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-03-30 11:32:13 +02:00
parent 3015008b1c
commit 464598773d
1 changed files with 31 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
nodes.py
View File

@ -1402,13 +1402,16 @@ class ImageConcanate:
{ {
"default": 'right' "default": 'right'
}), }),
"match_image_size": ("BOOLEAN", {"default": True}),
}} }}
RETURN_TYPES = ("IMAGE",) RETURN_TYPES = ("IMAGE",)
FUNCTION = "concanate" FUNCTION = "concanate"
CATEGORY = "KJNodes" CATEGORY = "KJNodes"
def concanate(self, image1, image2, direction): def concanate(self, image1, image2, direction, match_image_size):
if match_image_size:
image2 = torch.nn.functional.interpolate(image2, size=(image1.shape[2], image1.shape[3]), mode="bilinear")
if direction == 'right': if direction == 'right':
row = torch.cat((image1, image2), dim=2) row = torch.cat((image1, image2), dim=2)
elif direction == 'down': elif direction == 'down':
@ -3908,6 +3911,30 @@ class Superprompt:
return (out, ) return (out, )
class RemapImageRange:
@classmethod
def INPUT_TYPES(s):
return {"required": {
"image": ("IMAGE",),
"min": ("FLOAT", {"default": 0.0,"min": -10.0, "max": 1.0, "step": 0.01}),
"max": ("FLOAT", {"default": 1.0,"min": 0.0, "max": 10.0, "step": 0.01}),
"clamp": ("BOOLEAN", {"default": True}),
},
}
RETURN_TYPES = ("IMAGE",)
FUNCTION = "remap"
CATEGORY = "Marigold"
def remap(self, image, min, max, clamp):
if image.dtype == torch.float16:
image = image.to(torch.float32)
image = min + image * (max - min)
if clamp:
image = torch.clamp(image, min=0.0, max=1.0)
return (image, )
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {
"INTConstant": INTConstant, "INTConstant": INTConstant,
"FloatConstant": FloatConstant, "FloatConstant": FloatConstant,
@ -3978,7 +4005,8 @@ NODE_CLASS_MAPPINGS = {
"Intrinsic_lora_sampling": Intrinsic_lora_sampling, "Intrinsic_lora_sampling": Intrinsic_lora_sampling,
"RemapMaskRange": RemapMaskRange, "RemapMaskRange": RemapMaskRange,
"LoadResAdapterNormalization": LoadResAdapterNormalization, "LoadResAdapterNormalization": LoadResAdapterNormalization,
"Superprompt": Superprompt "Superprompt": Superprompt,
"RemapImageRange": RemapImageRange
} }
NODE_DISPLAY_NAME_MAPPINGS = { NODE_DISPLAY_NAME_MAPPINGS = {
"INTConstant": "INT Constant", "INTConstant": "INT Constant",
@ -4050,4 +4078,5 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"RemapMaskRange": "RemapMaskRange", "RemapMaskRange": "RemapMaskRange",
"LoadResAdapterNormalization": "LoadResAdapterNormalization", "LoadResAdapterNormalization": "LoadResAdapterNormalization",
"Superprompt": "Superprompt", "Superprompt": "Superprompt",
"RemapImageRange": "RemapImageRange",
} }