| """模型架构模块 (Person B 负责)""" | |
| from easytranslate.model.transformer import TransformerTranslationModel | |
| from easytranslate.model.encoder import TransformerEncoder | |
| from easytranslate.model.decoder import TransformerDecoder | |
| from easytranslate.model.attention import MultiHeadAttention, FlashMultiHeadAttention | |
| from easytranslate.model.positional import SinusoidalPositionalEncoding, RotaryPositionalEmbedding | |
| from easytranslate.model.finetune import load_pretrained_model, setup_lora | |
| __all__ = [ | |
| "TransformerTranslationModel", | |
| "TransformerEncoder", | |
| "TransformerDecoder", | |
| "MultiHeadAttention", | |
| "FlashMultiHeadAttention", | |
| "SinusoidalPositionalEncoding", | |
| "RotaryPositionalEmbedding", | |
| "load_pretrained_model", | |
| "setup_lora", | |
| ] | |