This commit is contained in:
Yousef Rafat 2026-05-08 12:31:18 +03:00
parent 7b89cd587c
commit 29452cd103

View File

@ -20,8 +20,8 @@ class LoadBackgroundRemovalModel(IO.ComfyNode):
] ]
) )
@classmethod @classmethod
def execute(cls, background_removal_name): def execute(cls, bg_removal_name):
path = folder_paths.get_full_path_or_raise("background_removal", background_removal_name) path = folder_paths.get_full_path_or_raise("background_removal", bg_removal_name)
bg = load(path) bg = load(path)
if bg is None: if bg is None:
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.")