Update custom_pipeline/WanTransformer3DModel.py

custom_pipeline/WanTransformer3DModel.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import math
+from typing import Optional, Tuple, Union, Dict, Any  # Optional과 기타 필요한 타입 힌트 추가
 from diffusers.configuration_utils import ConfigMixin, register_to_config
 from diffusers.models import ModelMixin
 from diffusers.loaders import FromOriginalModelMixin, PeftAdapterMixin
@@ -21,7 +22,7 @@ from diffusers.models.attention import FeedForward
 from diffusers.models.attention_processor import Attention
 from diffusers.models.embeddings import PixArtAlphaTextProjection, TimestepEmbedding, Timesteps, get_1d_rotary_pos_embed
 from diffusers.models.modeling_outputs import Transformer2DModelOutput
-from diffusers.models.modeling_utils import ModelMixin as ModelMixinBase  # ModelMixin 중복 방지
+from diffusers.models.modeling_utils import ModelMixin as ModelMixinBase
 from diffusers.models.normalization import FP32LayerNorm
 
 import torch