Spaces:
Running
on
T4
Running
on
T4
More debugging printing
Browse files
models/GroundingDINO/groundingdino.py
CHANGED
@@ -598,6 +598,7 @@ class GroundingDINO(nn.Module):
|
|
598 |
poss.append(pos_l)
|
599 |
|
600 |
input_query_bbox = input_query_label = attn_mask = dn_meta = None
|
|
|
601 |
hs, reference, hs_enc, ref_enc, init_box_proposal = self.transformer(
|
602 |
srcs, masks, input_query_bbox, poss, input_query_label, attn_mask, text_dict
|
603 |
)
|
|
|
598 |
poss.append(pos_l)
|
599 |
|
600 |
input_query_bbox = input_query_label = attn_mask = dn_meta = None
|
601 |
+
print("passing info through transformer")
|
602 |
hs, reference, hs_enc, ref_enc, init_box_proposal = self.transformer(
|
603 |
srcs, masks, input_query_bbox, poss, input_query_label, attn_mask, text_dict
|
604 |
)
|