damerajee commited on
Commit
e2a90a8
·
verified ·
1 Parent(s): 366c6df

Update modeling_gpt2vision.py

Browse files
Files changed (1) hide show
  1. modeling_gpt2vision.py +1 -1
modeling_gpt2vision.py CHANGED
@@ -84,7 +84,7 @@ class GPT2Vision(PreTrainedModel):
84
  }
85
  print("batch",batch)
86
  print("pixel_values",batch['pixel_values'].size())
87
- print("input_ids",batch['input_ids'[.size())
88
  print("attention_mask",batch['attention_mask'].size())
89
 
90
  inputs_embeds, attention_mask, input_ids = self.preprocess_inputs(batch)
 
84
  }
85
  print("batch",batch)
86
  print("pixel_values",batch['pixel_values'].size())
87
+ print("input_ids",batch['input_ids'].size())
88
  print("attention_mask",batch['attention_mask'].size())
89
 
90
  inputs_embeds, attention_mask, input_ids = self.preprocess_inputs(batch)