metacontroller-pytorch 0.0.23__py3-none-any.whl → 0.0.24__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 (6) hide show
  1. metacontroller/metacontroller.py +3 -0
  2. {metacontroller_pytorch-0.0.23.dist-info → metacontroller_pytorch-0.0.24.dist-info}/METADATA +3 -3
  3. metacontroller_pytorch-0.0.24.dist-info/RECORD +6 -0
  4. metacontroller_pytorch-0.0.23.dist-info/RECORD +0 -6
  5. {metacontroller_pytorch-0.0.23.dist-info → metacontroller_pytorch-0.0.24.dist-info}/WHEEL +0 -0
  6. {metacontroller_pytorch-0.0.23.dist-info → metacontroller_pytorch-0.0.24.dist-info}/licenses/LICENSE +0 -0
metacontroller/metacontroller.py CHANGED
@@ -27,6 +27,7 @@ from discrete_continuous_embed_readout import Embed, Readout, EmbedAndReadout
27
27
  from assoc_scan import AssocScan
28
28
 
29
29
  from torch_einops_utils import pad_at_dim
30
+ from torch_einops_utils.save_load import save_load
30
31
 
31
32
  # constants
32
33
 
@@ -63,6 +64,7 @@ MetaControllerOutput = namedtuple('MetaControllerOutput', (
63
64
  'switch_loss'
64
65
  ))
65
66
 
67
+ @save_load()
66
68
  class MetaController(Module):
67
69
  def __init__(
68
70
  self,
@@ -272,6 +274,7 @@ TransformerOutput = namedtuple('TransformerOutput', (
272
274
  'prev_hiddens'
273
275
  ))
274
276
 
277
+ @save_load()
275
278
  class Transformer(Module):
276
279
  def __init__(
277
280
  self,
{metacontroller_pytorch-0.0.23.dist-info → metacontroller_pytorch-0.0.24.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: metacontroller-pytorch
3
- Version: 0.0.23
3
+ Version: 0.0.24
4
4
  Summary: Transformer Metacontroller
5
5
  Project-URL: Homepage, https://pypi.org/project/metacontroller/
6
6
  Project-URL: Repository, https://github.com/lucidrains/metacontroller
@@ -39,8 +39,8 @@ Requires-Dist: discrete-continuous-embed-readout>=0.1.12
39
39
  Requires-Dist: einops>=0.8.1
40
40
  Requires-Dist: einx>=0.3.0
41
41
  Requires-Dist: loguru
42
- Requires-Dist: memmap-replay-buffer>=0.0.1
43
- Requires-Dist: torch-einops-utils>=0.0.7
42
+ Requires-Dist: memmap-replay-buffer>=0.0.23
43
+ Requires-Dist: torch-einops-utils>=0.0.16
44
44
  Requires-Dist: torch>=2.5
45
45
  Requires-Dist: x-evolution>=0.1.23
46
46
  Requires-Dist: x-mlps-pytorch
metacontroller_pytorch-0.0.24.dist-info/RECORD ADDED
@@ -0,0 +1,6 @@
1
+ metacontroller/__init__.py,sha256=lj7IOGpN--qMxJWbB-4SGqoPXG7Hd4mgtToTRSyTZ58,57
2
+ metacontroller/metacontroller.py,sha256=Ocm_2hCBvV2coYg4tQ4kYd0LQHgHWiz1l-c9lR7Z_fM,13941
3
+ metacontroller_pytorch-0.0.24.dist-info/METADATA,sha256=5xKHBecV3iRSK-JbNwOQ0iv6KSz_sIN--ar_M05-EWQ,4363
4
+ metacontroller_pytorch-0.0.24.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
5
+ metacontroller_pytorch-0.0.24.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
6
+ metacontroller_pytorch-0.0.24.dist-info/RECORD,,
metacontroller_pytorch-0.0.23.dist-info/RECORD DELETED
@@ -1,6 +0,0 @@
1
- metacontroller/__init__.py,sha256=lj7IOGpN--qMxJWbB-4SGqoPXG7Hd4mgtToTRSyTZ58,57
2
- metacontroller/metacontroller.py,sha256=mxXLeUTJ2AeYYUzpJQeE3TOrtZG_D45vfkb3jTNeEb8,13864
3
- metacontroller_pytorch-0.0.23.dist-info/METADATA,sha256=md2Ew7uonopeMK6YFkgfcXrs_3AoIQ_5gzu9Om3Fhuc,4361
4
- metacontroller_pytorch-0.0.23.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
5
- metacontroller_pytorch-0.0.23.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
6
- metacontroller_pytorch-0.0.23.dist-info/RECORD,,