Spaces:

dragonSwing
/

capu-demo

Runtime error

dragonSwing commited on May 17, 2022

Commit

7fbb729

•

1 Parent(s): 87ce977

Fix device bug

Files changed (1) hide show

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),
         )