diff --git a/xlstm/blocks/slstm/layer.py b/xlstm/blocks/slstm/layer.py index 2f3eb7d..2502db1 100644 --- a/xlstm/blocks/slstm/layer.py +++ b/xlstm/blocks/slstm/layer.py @@ -103,7 +103,7 @@ def step( else: x_conv = x - i, f, z, o = ( + f, i, z, o = ( self.fgate(x_conv), self.igate(x_conv), self.zgate(x),