Upload image_processing_vqmodel.py

image_processing_vqmodel.py

@@ -51,7 +51,8 @@ class VQModelImageProcessor(BaseImageProcessor):  # type: ignore
         width: int | None = None,
         height: int | None = None,
     ) -> Image.Image:
-        x_np = x.numpy().transpose(1, 2, 0)
+        x_np = x.detach().cpu().numpy()
+        x_np = x_np.transpose(1, 2, 0)
         x_np = (x_np + 1.0) * 127.5
         x_np = np.clip(x_np, 0, 255).astype(np.uint8)
         image = Image.fromarray(x_np)