chore: 更新项目名称

tests/module/test_module.py

@@ -5,11 +5,11 @@ import shutil
 import pytest
 import tempfile
 import safetensors.torch as st
-from khaosz.trainer import *
-from khaosz.config import *
-from khaosz.model import *
-from khaosz.data import *
-from khaosz.inference.generator import EmbeddingEncoderCore, GeneratorCore
+from astrai.trainer import *
+from astrai.config import *
+from astrai.model import *
+from astrai.data import *
+from astrai.inference.generator import EmbeddingEncoderCore, GeneratorCore
 from tokenizers import pre_tokenizers
 
 

tests/module/test_tie_weight.py

@@ -4,8 +4,8 @@ import torch
 import pytest
 import tempfile
 import safetensors.torch as st
-from khaosz.model.transformer import Transformer
-from khaosz.config.model_config import ModelConfig
+from astrai.model.transformer import Transformer
+from astrai.config.model_config import ModelConfig
 
 
 @pytest.fixture