Transformer_564M / modules.py
yagizdevre's picture
added configs
6ff2080
raw
history blame contribute delete
104 Bytes
from .attn import Attention
from .attn import AttentionSDPA
from .mlp import MLP
from .stu import STU