refactor: 修改模型架构

This commit is contained in:
2026-04-16 18:57:21 +08:00
parent cb93f8219e
commit a38334f4ce
17 changed files with 540 additions and 176 deletions
+2 -2
View File
@@ -2,7 +2,7 @@
from typing import Dict, List, Type
from pipeline.tokenizer import BpeTokenizer
from pipeline.tokenize import AutoTokenizer
from pipeline.strategies.base import PromptStrategy
@@ -44,7 +44,7 @@ class StrategyFactory:
return decorator
@classmethod
def create(cls, name: str, tokenizer: BpeTokenizer, **kwargs) -> PromptStrategy:
def create(cls, name: str, tokenizer: AutoTokenizer, **kwargs) -> PromptStrategy:
"""Create a strategy by name.
Args: