EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

[Model] Support ArcticForCausalLM architecture (Snowflake/snowflake-arctic-instruct) (#5078)

Browse Source 
Co-authored-by: vincent-4 <vincentzhongy+githubvincent4@gmail.com>
This commit is contained in:
Brayden Zhong
2025-04-25 03:24:09 -04:00
committed by GitHub
parent 7d9679b74d
commit 43fb95c2fa
5 changed files with 766 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/sglang/srt/hf_transformers_utils.py
View File

@@ -31,6 +31,7 @@ from transformers import (
from transformers.models.auto.modeling_auto import MODEL_FOR_CAUSAL_LM_MAPPING_NAMES
from sglang.srt.configs import (
ArcticConfig,
ChatGLMConfig,
DbrxConfig,
DeepseekVL2Config,
@@ -41,6 +42,7 @@ from sglang.srt.connector import create_remote_connector
from sglang.srt.utils import is_remote_url
_CONFIG_REGISTRY: Dict[str, Type[PretrainedConfig]] = {
ArcticConfig.model_type: ArcticConfig,
ChatGLMConfig.model_type: ChatGLMConfig,
DbrxConfig.model_type: DbrxConfig,
ExaoneConfig.model_type: ExaoneConfig,