wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-14 11:07:24 +08:00
Code Issues Actions 26 Packages Projects Releases Wiki Activity

name updates

Browse Source
This commit is contained in:
Yousef Rafat 2026-05-08 12:06:10 +03:00
parent 640109efb3
commit e976529c52
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy_extras/nodes_bg_removal.py
View File

@ -4,12 +4,12 @@ from comfy_api.latest import ComfyExtension, IO
from comfy.bg_removal_model import load from comfy.bg_removal_model import load
class LoadBackGroundRemovalModel(IO.ComfyNode): class LoadBackgroundRemovalModel(IO.ComfyNode):
@classmethod @classmethod
def define_schema(cls): def define_schema(cls):
files = folder_paths.get_filename_list("background_removal") files = folder_paths.get_filename_list("background_removal")
return IO.Schema( return IO.Schema(
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)),
@ -26,7 +26,7 @@ class LoadBackGroundRemovalModel(IO.ComfyNode):
raise RuntimeError("ERROR: background model file is invalid and does not contain a valid background removal model.") raise RuntimeError("ERROR: background model file is invalid and does not contain a valid background removal model.")
return IO.NodeOutput(bg) return IO.NodeOutput(bg)
class RemoveBackGround(IO.ComfyNode): class RemoveBackground(IO.ComfyNode):
@classmethod @classmethod
def define_schema(cls): def define_schema(cls):
return IO.Schema( return IO.Schema(