refactor: Set RELOAD_INST on all nodes

This commit is contained in:
Michael Poutre
2023-09-25 22:14:21 -07:00
parent 6a0d4123d1
commit 336f092d6f
+8
View File
@@ -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"