Upload processor

image_processing.py

@@ -22,7 +22,9 @@ class LSPDetrImageProcessor(BaseImageProcessorFast):
         """
         radial_distances = outputs["radial_distances"].expm1()
 
-        t = torch.linspace(0, 1, radial_distances.size(-1) + 1, device=self.device)[:-1]
+        t = torch.linspace(
+            0, 1, radial_distances.size(-1) + 1, device=radial_distances.device
+        )[:-1]
         cos = torch.cos(2 * torch.pi * t)
         sin = torch.sin(2 * torch.pi * t)