GPU support

ud.py

@@ -21,14 +21,14 @@ class UniversalDependenciesPipeline(TokenClassificationPipeline):
           s,e=m[i]
           m.insert(i+1,(s+len(j)-2,e))
           m[i]=(s,s+len(j)-2)
-    r["offset_mapping"]=torch.tensor([m])
+    r["offset_mapping"]=torch.tensor([m],device=self.device)
     r["sentence"]=sentence
     return r
   def _forward(self,model_inputs):
     import torch
     v=model_inputs["input_ids"][0].tolist()
     with torch.no_grad():
-      e=self.model(input_ids=torch.tensor([v[0:i]+[self.tokenizer.mask_token_id]+v[i+1:]+[j] for i,j in enumerate(v[1:-1],1)]))
+      e=self.model(input_ids=torch.tensor([v[0:i]+[self.tokenizer.mask_token_id]+v[i+1:]+[j] for i,j in enumerate(v[1:-1],1)],device=self.device))
     return {"logits":e.logits[:,1:-2,:],**model_inputs}
   def postprocess(self,model_outputs,**kwargs):
     import numpy