update

app.py

@@ -492,5 +492,4 @@ with gr.Blocks(
 
 
 if __name__ == "__main__":
-    demo.queue()
     demo.launch()

asset3d_gen/models/delight.py

@@ -12,7 +12,6 @@ from huggingface_hub import snapshot_download
 from PIL import Image
 from asset3d_gen.models.segment import RembgRemover
 
-
 __all__ = [
     "DelightingModel",
 ]
@@ -58,8 +57,8 @@ class DelightingModel(object):
         pipeline.set_progress_bar_config(disable=True)
 
         pipeline.to(self.device, torch.float16)
-        pipeline.enable_model_cpu_offload()
-        pipeline.enable_xformers_memory_efficient_attention()
+        # pipeline.enable_model_cpu_offload()
+        # pipeline.enable_xformers_memory_efficient_attention()
         self.pipeline = pipeline
 
     def recenter_image(

asset3d_gen/models/super_resolution.py

@@ -8,7 +8,6 @@ from huggingface_hub import snapshot_download
 from PIL import Image
 from asset3d_gen.data.utils import get_images_from_grid
 
-
 logging.basicConfig(
     format="%(asctime)s - %(levelname)s - %(message)s", level=logging.INFO
 )
@@ -34,7 +33,7 @@ class ImageStableSR:
             torch_dtype=torch.float16,
         ).to(device)
         self.up_pipeline_x4.set_progress_bar_config(disable=True)
-        self.up_pipeline_x4.enable_model_cpu_offload()
+        # self.up_pipeline_x4.enable_model_cpu_offload()
 
     def __call__(
         self,

asset3d_gen/models/text_model.py

@@ -75,7 +75,7 @@ def build_text2img_ip_pipeline(
     pipe.set_ip_adapter_scale([ref_scale])
 
     pipe = pipe.to(device)
-    pipe.enable_model_cpu_offload()
+    # pipe.enable_model_cpu_offload()
     # pipe.enable_xformers_memory_efficient_attention()
     # pipe.enable_vae_slicing()
 
@@ -106,7 +106,7 @@ def build_text2img_pipeline(
         force_zeros_for_empty_prompt=False,
     )
     pipe = pipe.to(device)
-    pipe.enable_model_cpu_offload()
+    # pipe.enable_model_cpu_offload()
     pipe.enable_xformers_memory_efficient_attention()
 
     return pipe

asset3d_gen/models/texture_model.py

@@ -86,6 +86,6 @@ def build_texture_gen_pipe(
         pipe.set_ip_adapter_scale([ip_adapt_scale])
 
     pipe = pipe.to(device)
-    pipe.enable_model_cpu_offload()
+    # pipe.enable_model_cpu_offload()
 
     return pipe

asset3d_gen/scripts/render_mv.py

@@ -15,7 +15,6 @@ from torchvision import transforms
 from asset3d_gen.data.datasets import Asset3dGenDataset
 from asset3d_gen.models.texture_model import build_texture_gen_pipe
 
-
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)