Add AFMoE model implementation (#13216)

This commit is contained in:
Raghav Ravishankar
2026-01-16 18:05:42 +05:30
committed by GitHub
parent 3355b6e21b
commit daea51385d
5 changed files with 740 additions and 0 deletions

View File

@@ -44,6 +44,7 @@ from transformers import (
from transformers.models.auto.modeling_auto import MODEL_FOR_CAUSAL_LM_MAPPING_NAMES
from sglang.srt.configs import (
AfmoeConfig,
ChatGLMConfig,
DbrxConfig,
DeepseekVL2Config,
@@ -71,6 +72,7 @@ from sglang.srt.utils import is_remote_url, logger, lru_cache_frozenset, mistral
from sglang.srt.utils.patch_tokenizer import patch_tokenizer
_CONFIG_REGISTRY: List[Type[PretrainedConfig]] = [
AfmoeConfig,
ChatGLMConfig,
DbrxConfig,
ExaoneConfig,