update

infer_api.py

@@ -123,6 +123,7 @@ class BkgRemover:
             session_infer_path, providers=providers,
         )
 
+    @spaces.GPU
     def remove_background(
         self,
         img: np.ndarray,
@@ -868,7 +869,6 @@ class InferCanonicalAPI:
 
         self.bkg_remover = BkgRemover()
 
-    @spaces.GPU
     def canonicalize(self, image, seed):
         generator = torch.Generator(device=device).manual_seed(seed)
         return inference(
@@ -877,7 +877,6 @@ class InferCanonicalAPI:
             use_noise=self.use_noise, noise_d=self.noise_d, crop=True, seed=seed, timestep=self.timestep
         )
 
-    @spaces.GPU
     def gen(self, img_input, seed=0):
         if np.array(img_input).shape[-1] == 4 and np.array(img_input)[..., 3].min() == 255:
             # convert to RGB