fix(load): restore LabelAttentionConfig object from loaded dict

used as a namespace after, so no converting it throws a bug

Author: meilame-tayebjee

torchTextClassifiers/torchTextClassifiers.py

@@ -671,6 +671,10 @@ def load(cls, path: Union[str, Path], device: str = "auto") -> "torchTextClassif
 
         # Reconstruct model_config
         model_config = ModelConfig.from_dict(metadata["model_config"])
+        if type(model_config.label_attention_config) is dict:
+            model_config.label_attention_config = LabelAttentionConfig(
+                **model_config.label_attention_config
+            )
 
         # Create instance
         instance = cls(