make style && make quality;

Author: lawrence-cj

scripts/convert_sana_to_diffusers.py

@@ -7,6 +7,7 @@
 
 import torch
 from accelerate import init_empty_weights
+from huggingface_hub import hf_hub_download, snapshot_download
 from termcolor import colored
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
@@ -19,7 +20,7 @@
 )
 from diffusers.models.modeling_utils import load_model_dict_into_meta
 from diffusers.utils.import_utils import is_accelerate_available
-from huggingface_hub import hf_hub_download, snapshot_download
+
 
 CTX = init_empty_weights if is_accelerate_available else nullcontext
 

src/diffusers/__init__.py

@@ -806,8 +806,8 @@
             PixArtSigmaPAGPipeline,
             PixArtSigmaPipeline,
             ReduxImageEncoder,
-            SanaPipeline,
             SanaPAGPipeline,
+            SanaPipeline,
             SemanticStableDiffusionPipeline,
             ShapEImg2ImgPipeline,
             ShapEPipeline,

src/diffusers/models/transformers/sana_transformer.py

@@ -11,7 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import numbers
 from typing import Any, Dict, Optional, Union
 
 import torch
@@ -24,10 +23,7 @@
     Attention,
     AttentionProcessor,
     AttnProcessor2_0,
-    FusedAttnProcessor2_0,
     SanaLinearAttnProcessor2_0,
-    SanaMultiscaleAttnProcessor2_0,
-    SanaMultiscaleLinearAttention,
 )
 from ..embeddings import PatchEmbed, PixArtAlphaTextProjection, SinusoidalPositionalEmbedding
 from ..modeling_outputs import Transformer2DModelOutput
@@ -491,7 +487,7 @@ def forward(
         return_dict: bool = True,
     ):
         """
-        The [`PixArtTransformer2DModel`] forward method.
+        The [`SanaTransformer2DModel`] forward method.
 
         Args:
             hidden_states (`torch.FloatTensor` of shape `(batch size, channel, height, width)`):

src/diffusers/pipelines/__init__.py

@@ -592,8 +592,8 @@
             HunyuanDiTPAGPipeline,
             KolorsPAGPipeline,
             PixArtSigmaPAGPipeline,
-            StableDiffusion3PAGImg2ImgPipeline,
             SanaPAGPipeline,
+            StableDiffusion3PAGImg2ImgPipeline,
             StableDiffusion3PAGPipeline,
             StableDiffusionControlNetPAGInpaintPipeline,
             StableDiffusionControlNetPAGPipeline,