diff --git a/llm/src/llm/core/decoder.py b/llm/src/llm/core/decoder.py index 5e48366..3f96215 100644 --- a/llm/src/llm/core/decoder.py +++ b/llm/src/llm/core/decoder.py @@ -3,6 +3,7 @@ import torch from .feed_forward import FeedForward from .multi_head_attention import MultiHeadAttention + class Decoder(nn.Module): """ Базовый автогерессивный блок-декодер трансформера (без кэша KV).