funasr/models/rwkv_bat/rwkv.py @@ -76,6 +76,7 @@ x = x + self.dropout_ffn(ffn) return x, state class RWKVDecoderLayer(torch.nn.Module): """RWKV module.
