chore: 更新项目名称

tools/benchmark.py

@@ -1,7 +1,7 @@
 import torch
 from typing import Dict, Any
 from dataclasses import dataclass
-from khaosz.model.transformer import ModelConfig, Transformer
+from astrai.model.transformer import ModelConfig, Transformer
 
 
 @dataclass

tools/generate.py

@@ -2,9 +2,9 @@ import torch
 import json
 import argparse
 
-from khaosz.config.param_config import ModelParameter
-from khaosz.inference.generator import BatchGenerator, GenerationRequest
-from khaosz.inference.core import disable_random_init
+from astrai.config.param_config import ModelParameter
+from astrai.inference.generator import BatchGenerator, GenerationRequest
+from astrai.inference.core import disable_random_init
 
 
 def processor(

tools/perplexity.py

@@ -6,8 +6,8 @@ import argparse
 import tqdm
 
 from torch import Tensor
-from khaosz.config.param_config import ModelParameter
-from khaosz.inference.core import disable_random_init
+from astrai.config.param_config import ModelParameter
+from astrai.inference.core import disable_random_init
 
 
 def compute_perplexity(

tools/train.py

@@ -6,10 +6,10 @@ import torch.optim as optim
 from torch.nn.parallel import DistributedDataParallel as DDP
 
 from functools import partial
-from khaosz.data import DatasetLoader
-from khaosz.config import ModelParameter, TrainConfig, CosineScheduleConfig
-from khaosz.trainer import Trainer, SchedulerFactory
-from khaosz.parallel import get_rank
+from astrai.data import DatasetLoader
+from astrai.config import ModelParameter, TrainConfig, CosineScheduleConfig
+from astrai.trainer import Trainer, SchedulerFactory
+from astrai.parallel import get_rank
 
 
 def parse_args() -> argparse.Namespace: