| """模型架构模块 (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", |
| ] |
|
|
