Spaces:
Running
on
Zero
Running
on
Zero
| # Copyright (c) Meta Platforms, Inc. and affiliates. | |
| # All rights reserved. | |
| # | |
| # This source code is licensed under the license found in the | |
| # LICENSE file in the root directory of this source tree. | |
| # from .attention import MemEffAttention | |
| from .block import Block | |
| from .layer_scale import LayerScale | |
| from .mlp import Mlp | |
| from .patch_embed import PatchEmbed | |
| from .rope import PositionGetter, RotaryPositionEmbedding2D | |
| from .swiglu_ffn import SwiGLUFFN, SwiGLUFFNFused | |
| __all__ = [ | |
| Mlp, | |
| PatchEmbed, | |
| SwiGLUFFN, | |
| SwiGLUFFNFused, | |
| Block, | |
| # MemEffAttention, | |
| LayerScale, | |
| PositionGetter, | |
| RotaryPositionEmbedding2D, | |
| ] | |