mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-22 01:11:57 +00:00
linting
This commit is contained in:
@@ -11,6 +11,7 @@ if TYPE_CHECKING or _diffusers_available:
|
|||||||
from diffusers.configuration_utils import ConfigMixin, register_to_config
|
from diffusers.configuration_utils import ConfigMixin, register_to_config
|
||||||
from diffusers.models.modeling_utils import ModelMixin
|
from diffusers.models.modeling_utils import ModelMixin
|
||||||
else:
|
else:
|
||||||
|
|
||||||
class ModelMixin:
|
class ModelMixin:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
@@ -20,6 +21,7 @@ else:
|
|||||||
def register_to_config(init):
|
def register_to_config(init):
|
||||||
return init
|
return init
|
||||||
|
|
||||||
|
|
||||||
from .attention import flash_attention
|
from .attention import flash_attention
|
||||||
|
|
||||||
__all__ = ["WanModel"]
|
__all__ = ["WanModel"]
|
||||||
|
|||||||
Reference in New Issue
Block a user