From a1078af96d778266df67906f34fae8a8228f3cae Mon Sep 17 00:00:00 2001 From: Nikita Date: Tue, 12 Aug 2025 13:19:25 +0200 Subject: [PATCH 1/2] changed xLSTMRMS.. to RMS... --- src/transformers/models/xlstm/modeling_xlstm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/xlstm/modeling_xlstm.py b/src/transformers/models/xlstm/modeling_xlstm.py index 4dee0102e625..c57df1985088 100644 --- a/src/transformers/models/xlstm/modeling_xlstm.py +++ b/src/transformers/models/xlstm/modeling_xlstm.py @@ -31,7 +31,7 @@ if is_xlstm_available(): from xlstm.xlstm_large.model import mLSTMBlock as xLSTMBlock from xlstm.xlstm_large.model import mLSTMStateType, soft_cap - from xlstm.xlstm_large.model import xLSTMRMSNorm as xLSTMRMSNorm + from xlstm.xlstm_large.model import RMSNorm as xLSTMRMSNorm external_xlstm = True else: From 1c2adbe4c6a69cb116817ae1b6c00812a3f999b4 Mon Sep 17 00:00:00 2001 From: Nikita Date: Tue, 12 Aug 2025 18:50:35 +0200 Subject: [PATCH 2/2] fix linter error --- src/transformers/models/xlstm/modeling_xlstm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/xlstm/modeling_xlstm.py b/src/transformers/models/xlstm/modeling_xlstm.py index c57df1985088..0890279dc7df 100644 --- a/src/transformers/models/xlstm/modeling_xlstm.py +++ b/src/transformers/models/xlstm/modeling_xlstm.py @@ -29,9 +29,9 @@ if is_xlstm_available(): + from xlstm.xlstm_large.model import RMSNorm as xLSTMRMSNorm from xlstm.xlstm_large.model import mLSTMBlock as xLSTMBlock from xlstm.xlstm_large.model import mLSTMStateType, soft_cap - from xlstm.xlstm_large.model import RMSNorm as xLSTMRMSNorm external_xlstm = True else: