Add AFMoE model implementation (#13216)
This commit is contained in:
committed by
GitHub
parent
3355b6e21b
commit
daea51385d
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user