Skip to content

Commit 7264cd3

Browse files
committed
added support for from_single_file
1 parent 4aaa0d2 commit 7264cd3

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)