Allow one to set a custom progress callback (#2609)

ming030890

2025-08-05 b3fb4c0acd5f52a313f024b6f69b8f025c6eddfe

 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.