update

embodied_gen/models/sr_model.py

@@ -53,7 +53,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()
 
     @spaces.GPU
     def __call__(

embodied_gen/models/text_model.py

@@ -135,7 +135,7 @@ def build_text2img_ip_pipeline(
 
     pipe = pipe.to(device)
     pipe.image_encoder = pipe.image_encoder.to(device)
-    pipe.enable_model_cpu_offload()
+    # pipe.enable_model_cpu_offload()
     # pipe.enable_xformers_memory_efficient_attention()
     # pipe.enable_vae_slicing()
 
@@ -168,8 +168,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
 

embodied_gen/models/texture_model.py

@@ -106,6 +106,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