[Model] Add support for BharatGen's Param2MoE model (#38000)

Signed-off-by: bhargav-patel-29 <bhargav.patel@tihiitb.org> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
2026-04-06 13:49:56 +05:30
parent f6983f01de
commit c5e3454e5a
4 changed files with 906 additions and 0 deletions
--- a/tests/models/registry.py
+++ b/tests/models/registry.py
@@ -461,6 +461,10 @@ _TEXT_GENERATION_EXAMPLE_MODELS = {
        trust_remote_code=True,
        is_available_online=False,
    ),
+    "Param2MoEForCausalLM": _HfExamplesInfo(
+        "bharatgenai/Param2-17B-A2.4B-Thinking",
+        trust_remote_code=True,
+    ),
    "PersimmonForCausalLM": _HfExamplesInfo("adept/persimmon-8b-chat"),
    "PhiForCausalLM": _HfExamplesInfo("microsoft/phi-2"),
    "Phi3ForCausalLM": _HfExamplesInfo("microsoft/Phi-3-mini-4k-instruct"),