Spaces:

HorizonRobotics
/

EmbodiedGen-Texture-Gen

Running on Zero

xinjjj commited on Aug 18

Commit

14b4bbd

1 Parent(s): 509565d

update

Files changed (1) hide show

embodied_gen/models/text_model.py CHANGED Viewed

@@ -179,7 +179,7 @@ def build_text2img_ip_pipeline(
     pipe.set_ip_adapter_scale([ref_scale])
     if enable_cpu_offload:
-        # pipe.enable_model_cpu_offload()
     else:
         pipe = pipe.to(device)
         pipe.image_encoder = pipe.image_encoder.to(device)
@@ -242,7 +242,7 @@ def build_text2img_pipeline(
         force_zeros_for_empty_prompt=False,
     )
     if enable_cpu_offload:
-        # pipe.enable_model_cpu_offload()
     else:
         pipe = pipe.to(device)
     disable_xformers_flash3_on_blackwell()

     pipe.set_ip_adapter_scale([ref_scale])
     if enable_cpu_offload:
+        pass  # pipe.enable_model_cpu_offload()
     else:
         pipe = pipe.to(device)
         pipe.image_encoder = pipe.image_encoder.to(device)
         force_zeros_for_empty_prompt=False,
     )
     if enable_cpu_offload:
+        pass  # pipe.enable_model_cpu_offload()
     else:
         pipe = pipe.to(device)
     disable_xformers_flash3_on_blackwell()