refactor(config): 重命名 TransformerConfig 为 ModelConfig

This commit is contained in:
2025-11-07 07:31:12 +08:00
parent 66a551217e
commit 7e5ecf3b7d
9 changed files with 25 additions and 25 deletions
+2 -2
View File
@@ -9,7 +9,7 @@ import pytest
import matplotlib
from torch.utils.data import Dataset
from khaosz.config.model_config import TransformerConfig
from khaosz.config.model_config import ModelConfig
from khaosz.data.tokenizer import BpeTokenizer
from khaosz.model.transformer import Transformer
@@ -102,7 +102,7 @@ def base_test_env(request: pytest.FixtureRequest):
with open(config_path, 'w') as f:
json.dump(config, f)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
transformer_config = TransformerConfig().load(config_path)
transformer_config = ModelConfig().load(config_path)
model = Transformer(transformer_config).to(device=device)
tokenizer = BpeTokenizer()
+1 -1
View File
@@ -38,7 +38,7 @@ def test_env(request: pytest.FixtureRequest):
tokenizer.train_from_iterator(sp_token_iter, config["vocab_size"], 1)
tokenizer.save(tokenizer_path)
transformer_config = TransformerConfig().load(config_path)
transformer_config = ModelConfig().load(config_path)
model = Transformer(transformer_config)
st.save_file(model.state_dict(), model_path)