From dcaea7a280c2fcd6f08d4969600d9fd42fddf885 Mon Sep 17 00:00:00 2001 From: Jairo Correa Date: Sat, 14 Oct 2023 14:04:39 -0300 Subject: [PATCH] Fix get_extension_calling for single files --- comfy/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/utils.py b/comfy/utils.py index ecc3ea6f4..a9a8429c2 100644 --- a/comfy/utils.py +++ b/comfy/utils.py @@ -13,7 +13,7 @@ def get_extension_calling(): if "/custom_nodes/" in frame.filename: stack_module = inspect.getmodule(frame[0]) if stack_module: - return re.sub(r".*\.?custom_nodes\.([^\.]+).*", r"\1", stack_module.__name__).split(".")[0] + return re.sub(r".*\.?custom_nodes\.([^\.]+).*", r"\1", stack_module.__name__.replace("\\", ".").replace("/", ".")).split(".")[0] return None