expose internal model attribute:

Author: Gasoonjia

torchchat/model.py

@@ -470,13 +470,42 @@ def _replace_known_params(self, params):
                 params[key] = patterns[value]
         return params
 
-    def _load_model_state_dict(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs):
-        # 修改 state dict 中的键值
+    def _load_model_state_dict(
+        self,
+        state_dict,
+        prefix,
+        local_metadata,
+        strict,
+        missing_keys,
+        unexpected_keys,
+        error_msgs,
+    ):
+        # update key names to match the new model
         for key in list(state_dict.keys()):
-            new_key = 'model.' + key
+            new_key = "model." + key
             state_dict[new_key] = state_dict.pop(key)
         return state_dict
-    
+
+    def __getattr__(self, name):
+        """
+        Rewrite __getattr__ to search attribute in Model and its model attribute.
+        Note that this is a temporary solution to expose internal model attributes to the user.
+
+        :param name: The name of the attribute to get.
+        :return: The attribute value if found, otherwise raise AttributeError.
+        """
+        try:
+            return super().__getattribute__(name)
+        except AttributeError:
+            pass
+
+        try:
+            return getattr(self.model, name)
+        except AttributeError:
+            raise AttributeError(
+                f"'{type(self).__name__}' object has no attribute '{name}'"
+            )
+            
     @abstractmethod
     def forward(self, *args, **kwargs):
         raise NotImplementedError("forward method is not implemented")