Refactor conv1d call to ensure consistent state handling

NX-AI / xlstm

Official repository of the xLSTM.

GNU Affero General Public License v3.0

918 stars 66 forks source link

Open gukgurgel opened 5 days ago

gukgurgel commented 5 days ago

Switched the assignment of x_conv and conv_state based on return_last_state condition
Ensured conv_state is assigned when return_last_state is True