Fix TeleChatForCausalLM not register issue (#29473)

Signed-off-by: Lai, Yejing <yejing.lai@intel.com>
2025-11-26 21:15:00 +08:00
parent e30859dff3
commit bb706d6048
2 changed files with 5 additions and 0 deletions
--- a/tests/models/registry.py
+++ b/tests/models/registry.py
@@ -436,6 +436,9 @@ _TEXT_GENERATION_EXAMPLE_MODELS = {
    "SolarForCausalLM": _HfExamplesInfo(
        "upstage/solar-pro-preview-instruct", trust_remote_code=True
    ),
+    "TeleChatForCausalLM": _HfExamplesInfo(
+        "chuhac/TeleChat2-35B", trust_remote_code=True
+    ),
    "TeleChat2ForCausalLM": _HfExamplesInfo(
        "Tele-AI/TeleChat2-3B", trust_remote_code=True
    ),