diff --git a/src/diffusers/__init__.py b/src/diffusers/__init__.py index e848d1efea42..8924ea7b0df7 100644 --- a/src/diffusers/__init__.py +++ b/src/diffusers/__init__.py @@ -422,7 +422,7 @@ "ReduxImageEncoder", "SanaPAGPipeline", "SanaPipeline", - "SanaSCMPipeline", + "SanaSprintPipeline", "SemanticStableDiffusionPipeline", "ShapEImg2ImgPipeline", "ShapEPipeline", @@ -968,7 +968,7 @@ ReduxImageEncoder, SanaPAGPipeline, SanaPipeline, - SanaSCMPipeline, + SanaSprintPipeline, SemanticStableDiffusionPipeline, ShapEImg2ImgPipeline, ShapEPipeline, diff --git a/src/diffusers/pipelines/__init__.py b/src/diffusers/pipelines/__init__.py index 4504e6049143..50eb4672839f 100644 --- a/src/diffusers/pipelines/__init__.py +++ b/src/diffusers/pipelines/__init__.py @@ -280,7 +280,7 @@ _import_structure["paint_by_example"] = ["PaintByExamplePipeline"] _import_structure["pia"] = ["PIAPipeline"] _import_structure["pixart_alpha"] = ["PixArtAlphaPipeline", "PixArtSigmaPipeline"] - _import_structure["sana"] = ["SanaPipeline", "SanaSCMPipeline"] + _import_structure["sana"] = ["SanaPipeline", "SanaSprintPipeline"] _import_structure["semantic_stable_diffusion"] = ["SemanticStableDiffusionPipeline"] _import_structure["shap_e"] = ["ShapEImg2ImgPipeline", "ShapEPipeline"] _import_structure["stable_audio"] = [ @@ -651,7 +651,7 @@ from .paint_by_example import PaintByExamplePipeline from .pia import PIAPipeline from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline - from .sana import SanaPipeline, SanaSCMPipeline + from .sana import SanaPipeline, SanaSprintPipeline from .semantic_stable_diffusion import SemanticStableDiffusionPipeline from .shap_e import ShapEImg2ImgPipeline, ShapEPipeline from .stable_audio import StableAudioPipeline, StableAudioProjectionModel diff --git a/src/diffusers/pipelines/sana/__init__.py b/src/diffusers/pipelines/sana/__init__.py index 72f2402658ac..2d7dfde54f8b 100644 --- a/src/diffusers/pipelines/sana/__init__.py +++ b/src/diffusers/pipelines/sana/__init__.py @@ -23,7 +23,7 @@ _dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects)) else: _import_structure["pipeline_sana"] = ["SanaPipeline"] - _import_structure["pipeline_sana_scm"] = ["SanaSCMPipeline"] + _import_structure["pipeline_sana_scm"] = ["SanaSprintPipeline"] if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: try: @@ -34,7 +34,7 @@ from ...utils.dummy_torch_and_transformers_objects import * else: from .pipeline_sana import SanaPipeline - from .pipeline_sana_scm import SanaSCMPipeline + from .pipeline_sana_scm import SanaSprintPipeline else: import sys diff --git a/src/diffusers/pipelines/sana/pipeline_sana_scm.py b/src/diffusers/pipelines/sana/pipeline_sana_scm.py index cb5056598815..c45e8258204d 100644 --- a/src/diffusers/pipelines/sana/pipeline_sana_scm.py +++ b/src/diffusers/pipelines/sana/pipeline_sana_scm.py @@ -186,9 +186,9 @@ def retrieve_timesteps( return timesteps, num_inference_steps -class SanaSCMPipeline(DiffusionPipeline, SanaLoraLoaderMixin): +class SanaSprintPipeline(DiffusionPipeline, SanaLoraLoaderMixin): r""" - Pipeline for text-to-image generation using [Sana](https://huggingface.co/papers/2410.10629). + Pipeline for text-to-image generation using [SANA-Sprint](https://huggingface.co/papers/2503.09641). """ # fmt: off