impresso-project
/

ner-stacked-bert-multilingual

Token Classification

Model card Files Files and versions Community

emanuelaboros commited on Aug 9

Commit

f6fd959

•

1 Parent(s): 8c3be27

Update generic_ner.py

Files changed (1) hide show

generic_ner.py +0 -1

generic_ner.py CHANGED Viewed

@@ -235,7 +235,6 @@ class MultitaskTokenClassificationPipeline(Pipeline):
         input_ids = torch.tensor([inputs["input_ids"]], dtype=torch.long).to(
             self.model.device
         )
-        print(input_ids.shape)
         attention_mask = torch.tensor([inputs["attention_mask"]], dtype=torch.long).to(
             self.model.device
         )

         input_ids = torch.tensor([inputs["input_ids"]], dtype=torch.long).to(
             self.model.device
         )
         attention_mask = torch.tensor([inputs["attention_mask"]], dtype=torch.long).to(
             self.model.device
         )