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

2026-04-05 22:27:57 +08:00
parent 9b22b1651e
commit 39766aa1dc
21 changed files with 40 additions and 47 deletions
@@ -11,8 +11,8 @@ from tokenizers import pre_tokenizers
 from torch.utils.data import Dataset

 from astrai.config.model_config import ModelConfig
-from astrai.tokenize import BpeTokenizer, BpeTrainer
 from astrai.model.transformer import Transformer
+from astrai.tokenize import BpeTokenizer, BpeTrainer


 class RandomDataset(Dataset):