model: support JetVLM (#13289)

This commit is contained in:
Zijian Zhang
2025-11-18 12:02:03 +08:00
committed by GitHub
parent 9188feccca
commit aa8ecbda7a
9 changed files with 210 additions and 1 deletions

View File

@@ -32,6 +32,7 @@ import torch.distributed as dist
from sglang.srt.configs import (
FalconH1Config,
JetNemotronConfig,
JetVLMConfig,
KimiLinearConfig,
NemotronHConfig,
Qwen3NextConfig,
@@ -1416,7 +1417,7 @@ class ModelRunner:
@property
def hybrid_gdn_config(self):
config = self.model_config.hf_config
if isinstance(config, Qwen3NextConfig | JetNemotronConfig):
if isinstance(config, Qwen3NextConfig | JetNemotronConfig | JetVLMConfig):
return config
return None