model: Support Hybrid Mamba2 NemotronHForCausalLM (nvidia/NVIDIA-Nemotron-Nano-9B-v2) (#10909)
Signed-off-by: Netanel Haber <nhaber@nvidia.com>
This commit is contained in:
@@ -91,6 +91,11 @@ ALL_MODELS = [
|
||||
trust_remote_code=True,
|
||||
skip_long_prompt=True,
|
||||
),
|
||||
ModelCase(
|
||||
"nvidia/NVIDIA-Nemotron-Nano-9B-v2",
|
||||
trust_remote_code=True,
|
||||
skip_long_prompt=True,
|
||||
),
|
||||
ModelCase(
|
||||
"swiss-ai/Apertus-8B",
|
||||
trust_remote_code=True,
|
||||
|
||||
Reference in New Issue
Block a user