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:
co-authored by
sunjiaqi11
ispobock
parent
11b76d24dc
commit
eb4ba1bde2
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user