add Decorator for each condition

model.py

@@ -44,6 +44,7 @@ class Model:
         self.pipe = self.load_pipe(base_model_id, task_name)
         self.preprocessor = Preprocessor()
 
+    @spaces.GPU()
     def load_pipe(self, base_model_id: str, task_name) -> DiffusionPipeline:
         if (
             base_model_id == self.base_model_id
@@ -58,8 +59,8 @@ class Model:
             base_model_id, safety_checker=None, controlnet=controlnet, torch_dtype=torch.float32
         )
         pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
-        if self.device.type == "cuda":
-            pipe.disable_xformers_memory_efficient_attention()
+        # if self.device.type == "cuda":
+        #     pipe.disable_xformers_memory_efficient_attention()
         pipe.to(self.device)
 
         torch.cuda.empty_cache()
@@ -80,6 +81,7 @@ class Model:
             self.pipe = self.load_pipe(self.base_model_id, self.task_name)
         return self.base_model_id
 
+    @spaces.GPU()
     def load_controlnet_weight(self, task_name: str) -> None:
         if task_name == self.task_name:
             return