KeerthiVM commited on
Commit
5882944
·
1 Parent(s): 89a06bf
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -90,8 +90,9 @@ class Blip2QFormer(nn.Module):
90
 
91
  # Forward through BERT
92
  outputs = self.bert(
93
- None, # No text input
94
  attention_mask=None,
 
95
  encoder_hidden_states=visual_embeds,
96
  encoder_attention_mask=visual_attention_mask,
97
  return_dict=True
 
90
 
91
  # Forward through BERT
92
  outputs = self.bert(
93
+ input_ids=None, # No text input
94
  attention_mask=None,
95
+ inputs_embeds=query_tokens,
96
  encoder_hidden_states=visual_embeds,
97
  encoder_attention_mask=visual_attention_mask,
98
  return_dict=True