debug zerogpu

model.py

@@ -42,7 +42,7 @@ class Model:
     def __init__(
         self, base_model_id: str = "stable-diffusion-v1-5/stable-diffusion-v1-5", task_name: str = "Canny"
     ) -> None:
-        self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         self.base_model_id = ""
         self.task_name = ""
         self.pipe = self.load_pipe(base_model_id, task_name)
@@ -62,9 +62,9 @@ class Model:
             base_model_id, safety_checker=None, controlnet=controlnet, torch_dtype=torch.float16
         )
         pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
+        pipe.to(self.device)
         if self.device.type == "cuda":
             pipe.enable_xformers_memory_efficient_attention()
-        pipe.to(self.device)
         torch.cuda.empty_cache()
         gc.collect()
         self.base_model_id = base_model_id