[MODEL] Apertus and XIELU (#23068)

Signed-off-by: EduardDurech <39579228+EduardDurech@users.noreply.github.com>
Co-authored-by: AllenHaoHuang <allenhuangdd@gmail.com>
This commit is contained in:
EduardDurech
2025-08-29 14:29:18 +02:00
committed by GitHub
parent 4f7cde7272
commit 1cf3753b90
6 changed files with 696 additions and 1 deletions

View File

@@ -39,6 +39,7 @@ logger = init_logger(__name__)
# yapf: disable
_TEXT_GENERATION_MODELS = {
# [Decoder-only]
"ApertusForCausalLM": ("apertus", "ApertusForCausalLM"),
"AquilaModel": ("llama", "LlamaForCausalLM"),
"AquilaForCausalLM": ("llama", "LlamaForCausalLM"), # AquilaChat2
"ArceeForCausalLM": ("arcee", "ArceeForCausalLM"),