chore: 修改类名，优化导入顺序

astrai/model/__init__.py

@@ -1,3 +1,4 @@
+from astrai.model.automodel import AutoModel
 from astrai.model.module import (
     GQA,
     MLP,
@@ -6,8 +7,6 @@ from astrai.model.module import (
     RMSNorm,
 )
 from astrai.model.transformer import Transformer
-from astrai.model.automodel import AutoModel
-
 
 __all__ = [
     # Modules

astrai/model/automodel.py

@@ -2,12 +2,12 @@
 AutoModel base class for model loading and saving.
 """
 
-import torch.nn as nn
-import safetensors.torch as st
-
-from pathlib import Path
 from contextlib import contextmanager
-from typing import Self, Union, Dict, Type
+from pathlib import Path
+from typing import Dict, Self, Type, Union
+
+import safetensors.torch as st
+import torch.nn as nn
 
 from astrai.config import ModelConfig