tirex-mirror 2025.10.18__py3-none-any.whl → 2025.10.21__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (9) hide show
  1. tirex/base.py +3 -1
  2. tirex/models/slstm/cell.py +3 -1
  3. {tirex_mirror-2025.10.18.dist-info → tirex_mirror-2025.10.21.dist-info}/METADATA +1 -1
  4. {tirex_mirror-2025.10.18.dist-info → tirex_mirror-2025.10.21.dist-info}/RECORD +9 -9
  5. {tirex_mirror-2025.10.18.dist-info → tirex_mirror-2025.10.21.dist-info}/WHEEL +0 -0
  6. {tirex_mirror-2025.10.18.dist-info → tirex_mirror-2025.10.21.dist-info}/licenses/LICENSE +0 -0
  7. {tirex_mirror-2025.10.18.dist-info → tirex_mirror-2025.10.21.dist-info}/licenses/LICENSE_MIRROR.txt +0 -0
  8. {tirex_mirror-2025.10.18.dist-info → tirex_mirror-2025.10.21.dist-info}/licenses/NOTICE.txt +0 -0
  9. {tirex_mirror-2025.10.18.dist-info → tirex_mirror-2025.10.21.dist-info}/top_level.txt +0 -0
tirex/base.py CHANGED
@@ -76,7 +76,9 @@ class PretrainedModel(ABC):
76
76
  model = model.to(device)
77
77
 
78
78
  if compile and backend == "torch":
79
- sLSTMCellTorch.slstm_forward = torch.compile(sLSTMCellTorch.slstm_forward, mode="max-autotune")
79
+ compiled_slstm_forward = torch.compile(sLSTMCellTorch.slstm_forward)
80
+ for block in model.blocks:
81
+ block.slstm_layer.slstm_cell._impl_forward_torch = compiled_slstm_forward
80
82
  return model
81
83
 
82
84
  @classmethod
tirex/models/slstm/cell.py CHANGED
@@ -38,6 +38,8 @@ class sLSTMCell(nn.Module):
38
38
 
39
39
  self._bias_ = nn.Parameter(torch.empty((config.num_heads * config.num_gates * config.head_dim), dtype=None))
40
40
 
41
+ self._impl_forward_torch = sLSTMCellTorch.slstm_forward
42
+
41
43
  def forward(self, input: torch.Tensor, state: torch.Tensor) -> tuple[torch.Tensor, torch.Tensor]:
42
44
  input = self._get_input(input)
43
45
  state = self._get_state(input, state)
@@ -62,7 +64,7 @@ class sLSTMCell(nn.Module):
62
64
  .reshape(-1)
63
65
  )
64
66
 
65
- return sLSTMCellTorch.slstm_forward(input, state, recurrent_kernel, bias)
67
+ return self._impl_forward_torch(input, state, recurrent_kernel, bias)
66
68
 
67
69
  def _impl_cuda(self, input: torch.Tensor, state: torch.Tensor) -> torch.Tensor:
68
70
  if input.device.type != "cuda":
{tirex_mirror-2025.10.18.dist-info → tirex_mirror-2025.10.21.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: tirex-mirror
3
- Version: 2025.10.18
3
+ Version: 2025.10.21
4
4
  Summary: Unofficial mirror of NX-AI/tirex for packaging
5
5
  Author-email: Arpad Rozsas <rozsasarpi@gmail.com>
6
6
  License: NXAI COMMUNITY LICENSE AGREEMENT
{tirex_mirror-2025.10.18.dist-info → tirex_mirror-2025.10.21.dist-info}/RECORD RENAMED
@@ -1,5 +1,5 @@
1
1
  tirex/__init__.py,sha256=rfsOeCJ7eRqU3K3TOhfN5-4XUuZFqt11wBRxk5SoAWA,292
2
- tirex/base.py,sha256=JZuWzezuDenrlDekMaaPLBWzB9mw6vJ_qF2GWjou8ws,4229
2
+ tirex/base.py,sha256=cwmvqgvqHDFXkVZQ4qczPa6FPsp1c2SWvuByDR2Ecb8,4331
3
3
  tirex/util.py,sha256=AQ8eM-7IpGeAqegVqflrjPQITzIEKltRcJOBbJmikcA,22758
4
4
  tirex/api_adapter/__init__.py,sha256=YnTtPf5jGqvhfqoX8Ku7Yd0xohy0MmocE2ryrXVnQ1Q,135
5
5
  tirex/api_adapter/forecast.py,sha256=CpX9YfiQ1X6nvzODklCPzuKPlYidJDha8FnEU-6zr1Q,13919
@@ -10,12 +10,12 @@ tirex/models/__init__.py,sha256=YnTtPf5jGqvhfqoX8Ku7Yd0xohy0MmocE2ryrXVnQ1Q,135
10
10
  tirex/models/patcher.py,sha256=EOXFkHsPkq0nuxRNLAbnrgJtcYq0IMC3YIg_16WArg4,3213
11
11
  tirex/models/tirex.py,sha256=R21JEa6dJDZ8epGgOJcNIqDoCt12uLlg31JnBPed0Do,9815
12
12
  tirex/models/slstm/block.py,sha256=V91Amgz8WAOOHo4fK1UZxd4Dgbx4-X6kUBS6X4m0tKQ,2006
13
- tirex/models/slstm/cell.py,sha256=mZwR9qHMgRwsy4vLxAwd0SOEpq0Wo9MorPKBjHD9O9U,7259
13
+ tirex/models/slstm/cell.py,sha256=Q0eRInybFx7Mj_pmlCl3CW9SShT2Qxegrr9KZzy7nrU,7320
14
14
  tirex/models/slstm/layer.py,sha256=hrDydQJIAHf5W0A0Rt0hXG4yKXrOSY-HPL0UbigR6Q8,2867
15
- tirex_mirror-2025.10.18.dist-info/licenses/LICENSE,sha256=HlwHKnGTlE2oNm6734V-Vy62zlkWohnuZpYXSdkqDk4,7362
16
- tirex_mirror-2025.10.18.dist-info/licenses/LICENSE_MIRROR.txt,sha256=ulPZMcOZdN7JvISjiID3KUwovTjrPwiMv5ku9dM7nls,496
17
- tirex_mirror-2025.10.18.dist-info/licenses/NOTICE.txt,sha256=rcgDscFHb-uuZO3L0_vIxYhTYl-a2Rm0lBpp3_kKdFQ,147
18
- tirex_mirror-2025.10.18.dist-info/METADATA,sha256=rlXW7FPjN5fEqCWfuAcv14ewp7EEI8ucfV92nZXx1m8,11444
19
- tirex_mirror-2025.10.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
20
- tirex_mirror-2025.10.18.dist-info/top_level.txt,sha256=AOLDhfv0F_7nn3pFq0Kapg6Ky_28I_cGDXzQX3w9eO4,6
21
- tirex_mirror-2025.10.18.dist-info/RECORD,,
15
+ tirex_mirror-2025.10.21.dist-info/licenses/LICENSE,sha256=HlwHKnGTlE2oNm6734V-Vy62zlkWohnuZpYXSdkqDk4,7362
16
+ tirex_mirror-2025.10.21.dist-info/licenses/LICENSE_MIRROR.txt,sha256=ulPZMcOZdN7JvISjiID3KUwovTjrPwiMv5ku9dM7nls,496
17
+ tirex_mirror-2025.10.21.dist-info/licenses/NOTICE.txt,sha256=rcgDscFHb-uuZO3L0_vIxYhTYl-a2Rm0lBpp3_kKdFQ,147
18
+ tirex_mirror-2025.10.21.dist-info/METADATA,sha256=-AjoWsSw05eV19jrsN-v05jqHGc53j_-jnEix3ct6bE,11444
19
+ tirex_mirror-2025.10.21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
20
+ tirex_mirror-2025.10.21.dist-info/top_level.txt,sha256=AOLDhfv0F_7nn3pFq0Kapg6Ky_28I_cGDXzQX3w9eO4,6
21
+ tirex_mirror-2025.10.21.dist-info/RECORD,,