Add Jet-Nemotron (#12448)

This commit is contained in:
Zijian Zhang
2025-11-09 17:32:47 +08:00
committed by GitHub
parent 93cf60fc64
commit 3633f8b0cf
7 changed files with 678 additions and 2 deletions

View File

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