Feature/support longcat flash lite (#17838)

Co-authored-by: sunjiaqi11 <sunjiaqi11@meituan.com>
Co-authored-by: ispobock <ispobaoke@gmail.com>
This commit is contained in:
sjqgogogogo
2026-03-09 23:00:11 +08:00
committed by GitHub
co-authored by sunjiaqi11 ispobock
parent 11b76d24dc
commit eb4ba1bde2
16 changed files with 838 additions and 15 deletions
@@ -53,6 +53,9 @@ class LongcatFlashConfig(PretrainedConfig):
zero_expert_type="identity",
nextn_use_scmoe=False,
num_nextn_predict_layers=1,
ngram_vocab_size_ratio=None,
emb_neighbor_num=None,
emb_split_num=None,
**kwargs,
):
super().__init__(
@@ -102,3 +105,8 @@ class LongcatFlashConfig(PretrainedConfig):
self.zero_expert_type = zero_expert_type
self.routed_scaling_factor = routed_scaling_factor
self.hidden_act = "silu"
self.use_ngram_embedding = ngram_vocab_size_ratio is not None
if self.use_ngram_embedding:
self.ngram_embedding_m = int(ngram_vocab_size_ratio * vocab_size)
self.ngram_embedding_n = emb_neighbor_num
self.ngram_embedding_k = emb_split_num
@@ -193,6 +193,7 @@ class ModelConfig:
self.is_local_attention_model = is_local_attention_model(
self.hf_config.architectures
)
self.use_ngram_embedding = getattr(self.hf_config, "use_ngram_embedding", False)
self.is_piecewise_cuda_graph_disabled_model = (
is_piecewise_cuda_graph_disabled_model(self.hf_config.architectures)
or is_deepseek_nsa(self.hf_text_config)