refactor: Set RELOAD_INST on all nodes
This commit is contained in:
@@ -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"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user