debug zerogpu

model.py

@@ -108,8 +108,8 @@ class Model:
     def get_prompt(self, prompt: str, additional_prompt: str) -> str:
         return additional_prompt if not prompt else f"{prompt}, {additional_prompt}"
 
-    # @spaces.GPU #[uncomment to use ZeroGPU]
-    @torch.autocast("cuda")
+    @spaces.GPU #[uncomment to use ZeroGPU]
+    # @torch.autocast("cuda")
     def run_pipe(
         self,
         prompt: str,
@@ -131,7 +131,7 @@ class Model:
             image=control_image,
         ).images
 
-    @spaces.GPU #[uncomment to use ZeroGPU]
+    # @spaces.GPU #[uncomment to use ZeroGPU]
     @torch.inference_mode()
     def process_canny(
         self,