Update generic_ner.py

generic_ner.py

@@ -225,6 +225,7 @@ class MultitaskTokenClassificationPipeline(Pipeline):
         all_logits = {}
 
         for i in range(len(text_sentences)):
+            print(inputs["input_ids"][i].shape)
             input_ids = torch.tensor([inputs["input_ids"][i]], dtype=torch.long).to(
                 self.model.device
             )