This commit is contained in:
Yousef Rafat 2026-05-08 12:19:54 +03:00
parent e976529c52
commit 270768067a

View File

@ -12,7 +12,7 @@ class LoadBackgroundRemovalModel(IO.ComfyNode):
node_id="LoadBackgroundRemovalModel", node_id="LoadBackgroundRemovalModel",
category="loaders", category="loaders",
inputs=[ inputs=[
IO.Combo.Input("bg_removal_name", options=sorted(files)), IO.Combo.Input("bg_removal_name", options=sorted(files), tooltip="The model used to remove backgrounds from images"),
], ],
outputs=[ outputs=[
IO.BackgroundRemoval.Output("bg_model") IO.BackgroundRemoval.Output("bg_model")
@ -34,11 +34,11 @@ class RemoveBackground(IO.ComfyNode):
display_name="Remove Background", display_name="Remove Background",
category="image/background removal", category="image/background removal",
inputs=[ inputs=[
IO.Image.Input("image"), IO.Image.Input("image", tooltip="Input image to remove the background from"),
IO.BackgroundRemoval.Input("bg_removal_model") IO.BackgroundRemoval.Input("bg_removal_model", tooltip="Background removal model used to generate the mask")
], ],
outputs=[ outputs=[
IO.Mask.Output("mask") IO.Mask.Output("mask", tooltip="Generated foreground mask")
] ]
) )
@classmethod @classmethod