diff --git a/moondream.py b/moondream.py index 439d79a..ce52209 100644 --- a/moondream.py +++ b/moondream.py @@ -88,7 +88,7 @@ class Moondream: RETURN_NAMES = ("text",) FUNCTION = "interrogate" OUTPUT_NODE = False - CATEGORY = "ImageTools" + CATEGORY = "InstantStudio" def interrogate(self, image:torch.Tensor, prompt:str, separator:str, model_revision:str, temperature:float, device:str):