This commit is contained in:
Maxime Ellerbach
2026-06-18 14:22:42 +00:00
parent 2f12f58e33
commit 84d33bf066
@@ -11,6 +11,7 @@ if TYPE_CHECKING or _diffusers_available:
from diffusers.configuration_utils import ConfigMixin, register_to_config
from diffusers.models.modeling_utils import ModelMixin
else:
class ModelMixin:
pass
@@ -20,6 +21,7 @@ else:
def register_to_config(init):
return init
from .attention import flash_attention
__all__ = ["WanModel"]