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

scripts/tools/generate.py

@@ -3,9 +3,9 @@ import json
 
 import torch
 
+from astrai.inference import InferenceEngine
 from astrai.model import AutoModel
 from astrai.tokenize import AutoTokenizer
-from astrai.inference import InferenceEngine
 
 
 def processor(

scripts/tools/train.py

@@ -2,10 +2,10 @@ import argparse
 import os
 from functools import partial
 
+import safetensors.torch as st
 import torch
 import torch.nn as nn
 import torch.optim as optim
-import safetensors.torch as st
 from torch.nn.parallel import DistributedDataParallel as DDP
 
 from astrai.config import ModelConfig, TrainConfig