Skip to content

Commit

Permalink
Merge branch 'main' of github.com:bentoml/ComfyUI-IDL
Browse files Browse the repository at this point in the history
  • Loading branch information
frostming committed Nov 8, 2024
2 parents 591851a + 4c1b273 commit 9a3d2af
Showing 1 changed file with 19 additions and 1 deletion.
20 changes: 19 additions & 1 deletion nodes/nodes.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ def __ne__(self, __value: object) -> bool:


class OutputPath:
COLOR = (142, 36, 170)
OUTPUT_NODE = True

@classmethod
Expand Down Expand Up @@ -56,7 +57,10 @@ def save(self, filename, filename_prefix):


def get_save_image_path(
filename_prefix: str, output_dir: str, image_width=0, image_height=0
filename_prefix: str,
output_dir: str,
image_width=0,
image_height=0,
) -> tuple[str, str, int, str, str]:
def map_filename(filename: str) -> tuple[int, str]:
prefix_len = len(os.path.basename(filename_prefix))
Expand Down Expand Up @@ -92,6 +96,8 @@ def map_filename(filename: str) -> tuple[int, str]:


class OutputImage:
COLOR = (142, 36, 170)

def __init__(self):
self.output_dir = folder_paths.get_output_directory()
self.type = "output"
Expand Down Expand Up @@ -157,6 +163,8 @@ def save_images(


class ImageInput:
COLOR = (142, 36, 170)

@classmethod
def INPUT_TYPES(s):
input_dir = folder_paths.get_input_directory()
Expand Down Expand Up @@ -236,6 +244,8 @@ def VALIDATE_INPUTS(s, image):


class StringInput:
COLOR = (142, 36, 170)

@classmethod
def INPUT_TYPES(s):
return {
Expand All @@ -260,6 +270,8 @@ def VALIDATE_INPUTS(s, value):


class IntegerInput:
COLOR = (142, 36, 170)

@classmethod
def INPUT_TYPES(s):
return {
Expand All @@ -284,6 +296,8 @@ def VALIDATE_INPUTS(s, value):


class FloatInput:
COLOR = (142, 36, 170)

@classmethod
def INPUT_TYPES(s):
return {
Expand All @@ -308,6 +322,8 @@ def VALIDATE_INPUTS(s, value):


class BooleanInput:
COLOR = (142, 36, 170)

@classmethod
def INPUT_TYPES(s):
return {
Expand All @@ -332,6 +348,8 @@ def VALIDATE_INPUTS(s, value):


class PathInput:
COLOR = (142, 36, 170)

@classmethod
def INPUT_TYPES(s):
return {
Expand Down

0 comments on commit 9a3d2af

Please sign in to comment.