transformers
01be5b48 - adds: extra_repr() to MambaRMSNorm to include hidden size / size of weights in the layer (#32171)

Commit
1 year ago
adds: extra_repr() to MambaRMSNorm to include hidden size / size of weights in the layer (#32171) * adds: extra_repr() to MambaRMSNorm to include the hidden size of the layer * style fix with ruff:
Parents
Loading