diff --git a/nodes/instance_diffusion_tracking_prompt_node.py b/nodes/instance_diffusion_tracking_prompt_node.py index d7140fb..030d02e 100644 --- a/nodes/instance_diffusion_tracking_prompt_node.py +++ b/nodes/instance_diffusion_tracking_prompt_node.py @@ -7,7 +7,7 @@ class InstanceDiffusionTrackingPromptNode: def INPUT_TYPES(s): return {"required": {"conditioning": ("CONDITIONING", ), "clip": ("CLIP", ), - # "tracking": ("TRACKING", ), + "tracking": ("TRACKING", ), "positionnet": ("POSITIONNET", ), "fusers": ("FUSERS", ), "fusers_batch_size": ("INT", {"default": 5, "min": 0, "max": 10000, "step": 8}), @@ -18,7 +18,7 @@ class InstanceDiffusionTrackingPromptNode: CATEGORY = "instance/conditioning" - def append(self, conditioning, clip, fusers, positionnet, fusers_batch_size, text): + def append(self, conditioning, clip, tracking, fusers, positionnet, fusers_batch_size, text): # Get prompts and their class id and trakcer id prompt_pairs = extract_prompts(text)