refactor(khaosz): 重构项目结构

2025-10-18 13:56:59 +08:00
parent 8434c19923
commit c51b203fde
28 changed files with 423 additions and 423 deletions
@@ -9,9 +9,11 @@ import pytest
 import matplotlib
 from torch.utils.data import Dataset

-from khaosz.core import *
-from khaosz.trainer import *
-from khaosz.trainer.data_util import *
+from khaosz.config.model_config import TransformerConfig
+from khaosz.data.data_util import build_attention_mask, build_loss_mask
+from khaosz.data.tokenizer import BpeTokenizer
+from khaosz.model.transformer import Transformer
+

 matplotlib.use("Agg")