From 7ef5a876de25c7e2d9d8b6541f6924b9ce5ffb08 Mon Sep 17 00:00:00 2001 From: logtd Date: Thu, 22 Feb 2024 23:06:37 -0700 Subject: [PATCH] add tracking to prompt --- nodes/instance_diffusion_tracking_prompt_node.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)