diff --git a/nodes.py b/nodes.py index b8a56aa..c96fde6 100644 --- a/nodes.py +++ b/nodes.py @@ -51,6 +51,7 @@ class SAMLoader: } } + RELOAD_INST = True RETURN_TYPES = ("SAM_MODEL",) FUNCTION = "load_model" @@ -87,6 +88,7 @@ class InitSAMPredictor: } } + RELOAD_INST = True RETURN_TYPES = ("SAM_PREDICTOR",) FUNCTION = "init_predictor" @@ -108,6 +110,7 @@ class LoadDINOModel: } } + RELOAD_INST = True RETURN_TYPES = ("DINO_MODEL",) FUNCTION = "load_model" @@ -175,6 +178,7 @@ class DinoBoxes: } } + RELOAD_INST = True RETURN_TYPES = ("IMAGE", "DINO_BOXES") FUNCTION = "get_boxes" @@ -214,6 +218,7 @@ class SAMMaskFromBoxes: } } + RELOAD_INST = True RETURN_TYPES = ("MASK",) FUNCTION = "get_mask" @@ -251,6 +256,7 @@ class MaskToTrimap: } } + RELOAD_INST = True RETURN_TYPES = ("TRIMAP","MASK") FUNCTION = "get_trimap" @@ -299,6 +305,7 @@ class LoadVITMatteModel: } } + RELOAD_INST = True RETURN_TYPES = ("VIT_MATTE_MODEL",) FUNCTION = "load_model" @@ -320,6 +327,7 @@ class GenerateVITMatte: } } + RELOAD_INST = True RETURN_TYPES = ("IMAGE",) FUNCTION = "generate_matte"