chanfee commited on
Commit
d14f2b2
·
verified ·
1 Parent(s): c91d22a

Update utils/model.py

Browse files
Files changed (1) hide show
  1. utils/model.py +1 -0
utils/model.py CHANGED
@@ -419,6 +419,7 @@ class OwlViTForClassification(nn.Module):
419
  else:
420
  #DEUBUG:
421
  print(f"text_inputs_parts - input_ids: {text_inputs_parts['input_ids'].shape}. attention_mask : {text_inputs_parts['attention_mask'].shape}")
 
422
  text_embeds_parts = self.owlvit.get_text_features(**text_inputs_parts)
423
 
424
  # # Embed images and text queries
 
419
  else:
420
  #DEUBUG:
421
  print(f"text_inputs_parts - input_ids: {text_inputs_parts['input_ids'].shape}. attention_mask : {text_inputs_parts['attention_mask'].shape}")
422
+ print(f"position_embedding: {self.owlvit.position_embedding()}")
423
  text_embeds_parts = self.owlvit.get_text_features(**text_inputs_parts)
424
 
425
  # # Embed images and text queries