Spaces:

VAST-AI
/

MIDI-3D

Running on Zero

huanngzh commited on 4 days ago

Commit

3a618ed

1 Parent(s): 58bc4de

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -159,7 +159,7 @@ def split_rgb_mask(rgb_image, seg_image):
     return instance_rgbs, instance_masks, scene_rgbs
-@spaces.GPU()
 @torch.no_grad()
 @torch.autocast(device_type=DEVICE, dtype=torch.bfloat16)
 def run_segmentation(image_prompts: Any, polygon_refinement: bool) -> Image.Image:
@@ -190,7 +190,7 @@ def run_segmentation(image_prompts: Any, polygon_refinement: bool) -> Image.Imag
     return seg_map_pil
-@spaces.GPU()
 @torch.no_grad()
 @torch.autocast(device_type=DEVICE, dtype=torch.bfloat16)
 def run_generation(

     return instance_rgbs, instance_masks, scene_rgbs
+# @spaces.GPU()
 @torch.no_grad()
 @torch.autocast(device_type=DEVICE, dtype=torch.bfloat16)
 def run_segmentation(image_prompts: Any, polygon_refinement: bool) -> Image.Image:
     return seg_map_pil
+# @spaces.GPU()
 @torch.no_grad()
 @torch.autocast(device_type=DEVICE, dtype=torch.bfloat16)
 def run_generation(