Commit
•
f6fd959
1
Parent(s):
8c3be27
Update generic_ner.py
Browse files- generic_ner.py +0 -1
generic_ner.py
CHANGED
@@ -235,7 +235,6 @@ class MultitaskTokenClassificationPipeline(Pipeline):
|
|
235 |
input_ids = torch.tensor([inputs["input_ids"]], dtype=torch.long).to(
|
236 |
self.model.device
|
237 |
)
|
238 |
-
print(input_ids.shape)
|
239 |
attention_mask = torch.tensor([inputs["attention_mask"]], dtype=torch.long).to(
|
240 |
self.model.device
|
241 |
)
|
|
|
235 |
input_ids = torch.tensor([inputs["input_ids"]], dtype=torch.long).to(
|
236 |
self.model.device
|
237 |
)
|
|
|
238 |
attention_mask = torch.tensor([inputs["attention_mask"]], dtype=torch.long).to(
|
239 |
self.model.device
|
240 |
)
|