fix import

Author: ChrisLiu6

src/diffusers/models/transformers/transformer_z_image.py

@@ -20,12 +20,11 @@
 import torch.nn.functional as F
 from torch.nn.utils.rnn import pad_sequence
 
-from diffusers.models.normalization import RMSNorm
-
 from ...configuration_utils import ConfigMixin, register_to_config
 from ...loaders import FromOriginalModelMixin, PeftAdapterMixin
 from ...models.attention_processor import Attention
 from ...models.modeling_utils import ModelMixin
+from ...models.normalization import RMSNorm
 from ...utils.torch_utils import maybe_allow_in_graph
 from ..attention_dispatch import dispatch_attention_fn