Spaces:

AlaBoussoffara
/

Mini-Transformer

Sleeping

Mini-Transformer / src /mini_transformer /modules /__init__.py

organized code and set up chainlit for demos

2d52135 about 1 year ago

230 Bytes

from .decoder import TransformerDecoder
from .embedding import InputEmbedding
from .encoder import TransformerEncoder
from .lm_head import LMHead

__all__ = ["TransformerEncoder", "TransformerDecoder", "InputEmbedding", "LMHead"]

	from .decoder import TransformerDecoder
	from .embedding import InputEmbedding
	from .encoder import TransformerEncoder
	from .lm_head import LMHead

	__all__ = ["TransformerEncoder", "TransformerDecoder", "InputEmbedding", "LMHead"]