feat: 优化工厂模式的实现
This commit is contained in:
@@ -6,7 +6,7 @@ from jinja2 import Template
|
||||
from torch import Tensor
|
||||
|
||||
from astrai.config.param_config import ModelParameter
|
||||
from astrai.core.factory import BaseFactory
|
||||
from astrai.factory import BaseFactory
|
||||
from astrai.inference.core import EmbeddingEncoderCore, GeneratorCore, KVCacheManager
|
||||
|
||||
HistoryType = List[Tuple[str, str]]
|
||||
@@ -299,8 +299,6 @@ class GeneratorFactory(BaseFactory[GeneratorCore]):
|
||||
result = generator.generate(request)
|
||||
"""
|
||||
|
||||
_registry: Dict[str, type] = {}
|
||||
|
||||
@staticmethod
|
||||
def create(parameter: ModelParameter, request: GenerationRequest) -> GeneratorCore:
|
||||
"""Create a generator based on request characteristics.
|
||||
|
||||
Reference in New Issue
Block a user