First commit
This commit is contained in:
11
pkgs/xformers/factory/__init__.py
Normal file
11
pkgs/xformers/factory/__init__.py
Normal file
@@ -0,0 +1,11 @@
|
||||
from xformers.components import MultiHeadDispatchConfig # noqa
|
||||
from xformers.components.attention import AttentionConfig # noqa
|
||||
from xformers.components.feedforward import FeedforwardConfig # noqa
|
||||
from xformers.components.positional_embedding import PositionEmbeddingConfig # noqa
|
||||
|
||||
from .block_factory import xFormerDecoderBlock # noqa
|
||||
from .block_factory import xFormerDecoderConfig # noqa
|
||||
from .block_factory import xFormerEncoderBlock # noqa
|
||||
from .block_factory import xFormerEncoderConfig # noqa
|
||||
from .model_factory import xFormer, xFormerConfig # noqa
|
||||
from .weight_init import xFormerWeightInit # noqa
|
||||
Reference in New Issue
Block a user