debug zerogpu

model.py

@@ -109,7 +109,7 @@ class Model:
         return additional_prompt if not prompt else f"{prompt}, {additional_prompt}"
 
     @spaces.GPU #[uncomment to use ZeroGPU]
-    @torch.autocast("cuda")
+    # @torch.autocast("cuda")
     def run_pipe(
         self,
         prompt: str,
@@ -121,6 +121,7 @@ class Model:
         seed: int,
     ) -> list[PIL.Image.Image]:
         generator = torch.Generator().manual_seed(seed)
+        self.pipe.to(self.device)
         return self.pipe(
             prompt=prompt,
             negative_prompt=negative_prompt,