Add wordpiece_mask to default to bool tensor
@@ -185,7 +185,7 @@ class PretrainedTransformerIndexer(TokenIndexer):
if key == "type_ids":
padding_value = 0
mktensor = torch.LongTensor
- elif key == "mask":
+ elif key == "mask" or key == "wordpiece_mask":
padding_value = False
mktensor = torch.BoolTensor
elif len(val) > 0 and isinstance(val[0], bool):
