Skip to content

Commit 2757284

Browse files
authored
fix: add XLinear to MODEL_FILENAME_DICT for save/load support (#1469)
1 parent d7a9aef commit 2757284

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

neuralforecast/core.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@
5656
TSMixer,
5757
TSMixerx,
5858
VanillaTransformer,
59+
XLinear,
5960
iTransformer,
6061
xLSTM,
6162
)
@@ -202,6 +203,8 @@ def _insample_times(
202203
"autotimexer": TimeXer,
203204
"xlstm": xLSTM,
204205
"autoxlstm": xLSTM,
206+
"xlinear": XLinear,
207+
"autoxlinear": XLinear,
205208
}
206209

207210

0 commit comments

Comments
 (0)