i

apps/gradio_app.py

@@ -21,6 +21,7 @@ proj_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 sys.path.append(os.path.join(proj_dir))
 
 import tempfile
+import spaces
 
 import craftsman
 from craftsman.systems.base import BaseSystem
@@ -68,6 +69,7 @@ If you have any questions, feel free to open a discussion or contact us at <b>we
 model = None
 cached_dir = None
 
+@spaces.GPU
 def image2mesh(view_front: np.ndarray, 
                view_right: np.ndarray, 
                view_back: np.ndarray, 

apps/mv_models.py

@@ -16,6 +16,7 @@ from typing import Dict, Optional, Tuple, List
 from dataclasses import dataclass
 from .utils import *
 from huggingface_hub import hf_hub_download
+import spaces
 
 parent_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 
@@ -27,7 +28,8 @@ class GenMVImage(object):
         self.step = 50
         self.pipelines = {}
         self.device = device
-        
+    
+    @spaces.GPU
     def gen_image_from_crm(self, image):
         from .third_party.CRM.pipelines import TwoStagePipeline
         stage1_config = OmegaConf.load(f"{parent_dir}/apps/third_party/CRM/configs/nf7_v3_SNR_rd_size_stroke.yaml").config
@@ -50,6 +52,7 @@ class GenMVImage(object):
         mv_imgs = rt_dict["stage1_images"]
         return mv_imgs[5], mv_imgs[3], mv_imgs[2], mv_imgs[0]
     
+    @spaces.GPU
     def gen_image_from_mvdream(self, image, text):
         from .third_party.mvdream_diffusers.pipeline_mvdream import MVDreamPipeline
         if image is None:
@@ -94,6 +97,7 @@ class GenMVImage(object):
                     )
         return mv_imgs[1], mv_imgs[2], mv_imgs[3], mv_imgs[0]
     
+    @spaces.GPU
     def gen_image_from_wonder3d(self, image, crop_size):
         sys.path.append(f"{parent_dir}/apps/third_party/Wonder3D")
         from diffusers import DiffusionPipeline  # only tested on diffusers[torch]==0.19.3, may have conflicts with newer versions of diffusers