nodes/logtd-ComfyUI-InstanceDiffusion
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remove test code

Browse Source
This commit is contained in:
logtd
2024-02-22 23:05:57 -07:00
parent 5a8deeaa94
commit 4214bcb9cc
1 changed files with 13 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/instance_diffusion_tracking_prompt_node.py
+13 -20
View File
@@ -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 = []