Update models/loaders/matanyone_loader.py

models/loaders/matanyone_loader.py

@@ -14,6 +14,7 @@
 - New: EasyDict polyfill and conversion for cfg to fix 'dict no attribute' errors
 - New: Full default cfg from official config.json to fix 'mem_every' issues
 - Update: Disable memory propagation by setting mem_every=-1, max_mem_frames=0 to fix dim mismatch in fusion
+- Fix: Merge long_term overrides to preserve keys like count_usage
 """
 from __future__ import annotations
 import os
@@ -444,14 +445,14 @@ def load(self) -> Optional[Any]:
                 "long_term": {
                     "buffer_tokens": 2000,
                     "count_usage": True,
-                    "max_mem_frames": 0,  # Disable long-term memory
+                    "max_mem_frames": 10,
                     "max_num_tokens": 10000,
                     "min_mem_frames": 5,
                     "num_prototypes": 128
                 },
                 "max_internal_size": -1,
-                "max_mem_frames": 0,  # Disable short-term memory
-                "mem_every": -1,  # Disable memory updates
+                "max_mem_frames": 5,
+                "mem_every": 5,
                 "model": {
                     "aux_loss": {
                         "query": {
@@ -540,12 +541,24 @@ def load(self) -> Optional[Any]:
                 'mem_every': -1,
                 'max_mem_frames': 0,
                 'use_long_term': False,
-                'long_term': {
-                    'max_mem_frames': 0,
-                    'min_mem_frames': 0,
-                },
             }
             cfg.update(overrides)
+            # Merge long_term overrides without overwriting whole dict
+            if 'long_term' in cfg:
+                cfg['long_term'].update({
+                    'max_mem_frames': 0,
+                    'min_mem_frames': 0,
+                    'count_usage': False,  # Disable since memory off
+                })
+            else:
+                cfg['long_term'] = {
+                    'buffer_tokens': 2000,
+                    'count_usage': False,
+                    'max_mem_frames': 0,
+                    'max_num_tokens': 10000,
+                    'min_mem_frames': 0,
+                    'num_prototypes': 128
+                }
             # Convert to EasyDict for dot access
             cfg = EasyDict(cfg)
             # Inference core