refactor: 从data 模块分离tokenizer
This commit is contained in:
+2
-1
@@ -6,7 +6,8 @@ from astrai.config import (
|
||||
TrainConfig,
|
||||
)
|
||||
from astrai.factory import BaseFactory
|
||||
from astrai.data import BpeTokenizer, DatasetFactory
|
||||
from astrai.data import DatasetFactory
|
||||
from astrai.tokenizer import BpeTokenizer
|
||||
from astrai.inference.generator import (
|
||||
BatchGenerator,
|
||||
EmbeddingEncoder,
|
||||
|
||||
Reference in New Issue
Block a user