dragonSwing
/

xlm-roberta-capu

Token Classification

Inference Endpoints

Model card Files Files and versions Community

dragonSwing commited on May 17, 2022

Commit

0a50beb

•

1 Parent(s): 6fe5f76

Fix device bug

Files changed (1) hide show

modeling_seq2labels.py +1 -1

modeling_seq2labels.py CHANGED Viewed

@@ -119,5 +119,5 @@ class Seq2LabelsModel(BertPreTrainedModel):
             detect_logits=logits_d,
             hidden_states=outputs.hidden_states,
             attentions=outputs.attentions,
-            max_error_probability=torch.ones(logits.size(0)),
         )

             detect_logits=logits_d,
             hidden_states=outputs.hidden_states,
             attentions=outputs.attentions,
+            max_error_probability=torch.ones(logits.size(0), device=logits.device),
         )