update

asset3d_gen/models/delight_model.py

@@ -57,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/sr_model.py

@@ -33,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,8 +106,8 @@ def build_text2img_pipeline(
         force_zeros_for_empty_prompt=False,
     )
     pipe = pipe.to(device)
-    pipe.enable_model_cpu_offload()
-    pipe.enable_xformers_memory_efficient_attention()
+    # 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