Skip to content

Commit 060b86e

Browse files
nikitazuevblagoNikita
andauthored
changed xLSTMRMSNorm to RMSNorm (#40113)
* changed xLSTMRMS.. to RMS... * fix linter error --------- Co-authored-by: Nikita <[email protected]>
1 parent 849c377 commit 060b86e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/models/xlstm/modeling_xlstm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,9 @@
2929

3030

3131
if is_xlstm_available():
32+
from xlstm.xlstm_large.model import RMSNorm as xLSTMRMSNorm
3233
from xlstm.xlstm_large.model import mLSTMBlock as xLSTMBlock
3334
from xlstm.xlstm_large.model import mLSTMStateType, soft_cap
34-
from xlstm.xlstm_large.model import xLSTMRMSNorm as xLSTMRMSNorm
3535

3636
external_xlstm = True
3737
else:

0 commit comments

Comments
 (0)