mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-10 09:12:31 +08:00
fix
This commit is contained in:
parent
7b89cd587c
commit
29452cd103
@ -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.")
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user