[model] Add mamba2 and Falcon-H1 support. (#10988)
Co-authored-by: Younes Belkada <younes.belkada@tii.ae> Co-authored-by: Younes B <49240599+younesbelkada@users.noreply.github.com>
This commit is contained in:
@@ -1297,6 +1297,7 @@ class ModelRunner:
|
||||
return self.model_config.hf_config.architectures[0] in [
|
||||
"Qwen3NextForCausalLM",
|
||||
"Qwen3NextForCausalLMMTP",
|
||||
"FalconH1ForCausalLM",
|
||||
]
|
||||
|
||||
def set_num_token_hybrid(self):
|
||||
|
||||
Reference in New Issue
Block a user