Merge pull request #1791 from dtlzhuangz/zhuangzhong_dev

Shi Xian

2024-06-11 7a9c0414b685b6f2ad7538017e2a1215e1e76901

 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.