update

infer_api.py

@@ -761,7 +761,7 @@ class InferSlrmAPI:
 
         return mesh_fpath
 
-
+@spaces.GPU
 class InferMultiviewAPI:
     def __init__(self, config):
         parser = argparse.ArgumentParser()
@@ -795,8 +795,6 @@ class InferMultiviewAPI:
         im = self.image_transforms(im)
         return im
 
-
-    @spaces.GPU
     def gen(self, img, seed, num_levels):
         set_seed(seed)
         data = {}
@@ -814,6 +812,7 @@ class InferMultiviewAPI:
 
 
 class InferCanonicalAPI:
+    @spaces.GPU
     def __init__(self, config):
         self.config = config
         self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")