update

Author: yiyixuxu

src/diffusers/pipelines/components_manager.py

@@ -266,7 +266,7 @@ def get(self, names: Union[str, List[str]]):
         else:
             raise ValueError(f"Invalid type for names: {type(names)}")
 
-    def enable_auto_cpu_offload(self, device, memory_reserve_margin="3GB"):
+    def enable_auto_cpu_offload(self, device: Union[str, int, torch.device]="cuda", memory_reserve_margin="3GB"):
         for name, component in self.components.items():
             if isinstance(component, torch.nn.Module) and hasattr(component, "_hf_hook"):
                 remove_hook_from_module(component, recurse=True)

src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_modular.py

@@ -3020,6 +3020,11 @@ def description(self):
     ("decode", StableDiffusionXLAutoDecodeStep)
 ])
 
+AUTO_CORE_BLOCKS = OrderedDict([
+    ("before_denoise", StableDiffusionXLAutoBeforeDenoiseStep),
+    ("denoise", StableDiffusionXLAutoDenoiseStep),
+])
+
 
 SDXL_SUPPORTED_BLOCKS = {
     "text2img": TEXT2IMAGE_BLOCKS,