Completetinymodelraven Top -

class TinyRavenBlock(nn.Module): def __init__(self, dim): self.attn = EfficientLinearAttention(dim) self.conv = DepthwiseConv1d(dim, kernel_size=3) self.ffn = nn.Sequential(nn.Linear(dim, dim*2), nn.GELU(), nn.Linear(dim*2, dim)) self.norm1 = nn.LayerNorm(dim) self.norm2 = nn.LayerNorm(dim)