File size: 769 Bytes
c1a46f7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | """模型架构模块 (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",
]
|