diff --git a/src/transformers/models/xlstm/modeling_xlstm.py b/src/transformers/models/xlstm/modeling_xlstm.py index 4dee0102e625..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 xLSTMRMSNorm as xLSTMRMSNorm external_xlstm = True else: