Issue fix
Browse files
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
|