mirror of
https://github.com/pese-git/llm-arch-research.git
synced 2026-01-24 05:21:16 +00:00
fix: format code
This commit is contained in:
@@ -3,6 +3,7 @@ import torch
|
|||||||
from .feed_forward import FeedForward
|
from .feed_forward import FeedForward
|
||||||
from .multi_head_attention import MultiHeadAttention
|
from .multi_head_attention import MultiHeadAttention
|
||||||
|
|
||||||
|
|
||||||
class Decoder(nn.Module):
|
class Decoder(nn.Module):
|
||||||
"""
|
"""
|
||||||
Базовый автогерессивный блок-декодер трансформера (без кэша KV).
|
Базовый автогерессивный блок-декодер трансформера (без кэша KV).
|
||||||
|
|||||||
Reference in New Issue
Block a user