WwYc commited on
Commit
e48c23b
·
verified ·
1 Parent(s): d5812e2

Update generic.py

Browse files
Files changed (1) hide show
  1. generic.py +1 -1
generic.py CHANGED
@@ -77,7 +77,7 @@ class ModelUsage:
77
  self.image_boxes_len = features.shape[1]
78
  self.bboxes = output_dict.get("boxes")
79
  self.output = self.lxmert_vqa(
80
- input_ids=inputs.input_ids),
81
  attention_mask=inputs.attention_mask,
82
  visual_feats=features,
83
  visual_pos=normalized_boxes,
 
77
  self.image_boxes_len = features.shape[1]
78
  self.bboxes = output_dict.get("boxes")
79
  self.output = self.lxmert_vqa(
80
+ input_ids=inputs.input_ids,
81
  attention_mask=inputs.attention_mask,
82
  visual_feats=features,
83
  visual_pos=normalized_boxes,