Create mask on device

Author: rlouf

outlines/text/generate/regex.py

@@ -128,9 +128,7 @@ def create_proposal(
         masks = []
         for pstate in self.pstates:
             mask = torch.full(
-                (len(self.model.tokenizer.vocabulary),),
-                -math.inf,
-                device=self.model.device,
+                (len(self.model.tokenizer.vocabulary),), -math.inf, device=self.device
             )
 
             if pstate[1] > -1: