Add dict_class to save_pretrained config for local model loading

save_pretrained now includes the model class name in config.json,
enabling load_dictionary_model to determine the correct class when
loading from a local save_pretrained directory. _from_pretrained
strips dict_class before instantiation to avoid __init__ TypeError.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: Butanium

dictionary_learning/dictionary.py

@@ -162,6 +162,17 @@ def decode(self, f):
         """
         pass
 
+    def save_pretrained(self, *args, **kwargs):
+        """Save model with dict_class in config for local loading."""
+        self._hub_mixin_config["dict_class"] = type(self).__name__
+        return super().save_pretrained(*args, **kwargs)
+
+    @classmethod
+    def _from_pretrained(cls, *, model_id, **kwargs):
+        """Strip dict_class from config before instantiation."""
+        kwargs.pop("dict_class", None)
+        return super()._from_pretrained(model_id=model_id, **kwargs)
+
     @classmethod
     @abstractmethod
     def from_pretrained(