Skip to content

Commit 452d946

Browse files
committed
Update sana_transformer.py
1 parent abb5a58 commit 452d946

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/models/transformers/sana_transformer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
from torch import nn
1919

2020
from ...configuration_utils import ConfigMixin, register_to_config
21-
from ...loaders import PeftAdapterMixin
21+
from ...loaders import PeftAdapterMixin,FromOriginalModelMixin
2222
from ...utils import USE_PEFT_BACKEND, logging, scale_lora_layers, unscale_lora_layers
2323
from ..attention_processor import (
2424
Attention,
@@ -195,7 +195,7 @@ def forward(
195195
return hidden_states
196196

197197

198-
class SanaTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin):
198+
class SanaTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin,FromOriginalModelMixin):
199199
r"""
200200
A 2D Transformer model introduced in [Sana](https://huggingface.co/papers/2410.10629) family of models.
201201

0 commit comments

Comments
 (0)