cocktailpeanut commited on
Commit
8a0a352
1 Parent(s): ac87f27
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,7 +26,7 @@ from core.options import AllConfigs, Options
26
  from core.models import LGM
27
  from mvdream.pipeline_mvdream import MVDreamPipeline
28
 
29
- import spaces
30
 
31
  IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406)
32
  IMAGENET_DEFAULT_STD = (0.229, 0.224, 0.225)
@@ -100,7 +100,7 @@ pipe_image = pipe_image.to(device)
100
  bg_remover = rembg.new_session()
101
 
102
  # process function
103
- @spaces.GPU
104
  def process(input_image, prompt, prompt_neg='', input_elevation=0, input_num_steps=30, input_seed=42):
105
 
106
  # seed
 
26
  from core.models import LGM
27
  from mvdream.pipeline_mvdream import MVDreamPipeline
28
 
29
+ #import spaces
30
 
31
  IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406)
32
  IMAGENET_DEFAULT_STD = (0.229, 0.224, 0.225)
 
100
  bg_remover = rembg.new_session()
101
 
102
  # process function
103
+ #@spaces.GPU
104
  def process(input_image, prompt, prompt_neg='', input_elevation=0, input_num_steps=30, input_seed=42):
105
 
106
  # seed