DeepLink-org / DeepLinkExt

BSD 3-Clause "New" or "Revised" License
11 stars 0 forks source link

fix: fix rms norm #102

Closed POI-WX closed 6 months ago

POI-WX commented 6 months ago

Fix rms norm on the wrong shape of inv_rms tensor.