From 4214bcb9cc79a15e7ebcfe1bc801ba8a58df24f6 Mon Sep 17 00:00:00 2001 From: logtd Date: Thu, 22 Feb 2024 23:05:57 -0700 Subject: [PATCH] remove test code --- ...instance_diffusion_tracking_prompt_node.py | 33 ++++++++----------- 1 file changed, 13 insertions(+), 20 deletions(-) diff --git a/nodes/instance_diffusion_tracking_prompt_node.py b/nodes/instance_diffusion_tracking_prompt_node.py index 352b115..d7140fb 100644 --- a/nodes/instance_diffusion_tracking_prompt_node.py +++ b/nodes/instance_diffusion_tracking_prompt_node.py @@ -23,27 +23,20 @@ class InstanceDiffusionTrackingPromptNode: prompt_pairs = extract_prompts(text) # Go through prompt pairs, encode prompts, and join with positions from tracking - # position_conds = [] - # for tracker_id, class_id, prompt in prompt_pairs: - # _, cond_pooled = clip.encode_from_tokens( - # clip.tokenize(prompt), return_pooled=True) - # # A tracker_id of -1 means that it is prompting all instances of a single class - # if tracker_id != -1: - # position_cond = {'cond_pooled': cond_pooled, 'positions': - # tracking[class_id][tracker_id]} - # position_conds.append(position_cond) - # else: - # for tracker_id in tracking[class_id]: - # position_cond = {'cond_pooled': cond_pooled, - # 'positions': tracking[class_id][tracker_id]} - # position_conds.append(position_cond) - - prompt = "chicken" - _, cond_pooled = clip.encode_from_tokens( + position_conds = [] + for tracker_id, class_id, prompt in prompt_pairs: + _, cond_pooled = clip.encode_from_tokens( clip.tokenize(prompt), return_pooled=True) - - position_conds = [{'cond_pooled': cond_pooled, - 'positions': [[100, 100, 200, 200, 512, 512]]}] + # A tracker_id of -1 means that it is prompting all instances of a single class + if tracker_id != -1: + position_cond = {'cond_pooled': cond_pooled, 'positions': + tracking[class_id][tracker_id]} + position_conds.append(position_cond) + else: + for tracker_id in tracking[class_id]: + position_cond = {'cond_pooled': cond_pooled, + 'positions': tracking[class_id][tracker_id]} + position_conds.append(position_cond) # Add prompts+embeddings to the input conditionings cond_out = []