[CI] Fix broken CI (#11543)

46d43594 · Robert Shaw · GitHub · 81b979f2 · 46d43594
Unverified Commit 46d43594 authored Dec 26, 2024 by Robert Shaw Committed by GitHub Dec 26, 2024
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

tests/models/registry.py tests/models/registry.py +2 -0

No files found.
--- a/tests/models/registry.py
+++ b/tests/models/registry.py
@@ -61,6 +61,8 @@ _TEXT_GENERATION_EXAMPLE_MODELS = {
    "DeepseekForCausalLM": _HfExamplesInfo("deepseek-ai/deepseek-llm-7b-chat"),
    "DeepseekV2ForCausalLM": _HfExamplesInfo("deepseek-ai/DeepSeek-V2-Lite-Chat",  # noqa: E501
                                         trust_remote_code=True),
+    "DeepseekV3ForCausalLM": _HfExamplesInfo("deepseek-ai/DeepSeek-V3",  # noqa: E501
+                                         trust_remote_code=True),
    "ExaoneForCausalLM": _HfExamplesInfo("LGAI-EXAONE/EXAONE-3.0-7.8B-Instruct"),  # noqa: E501
    "FalconForCausalLM": _HfExamplesInfo("tiiuae/falcon-7b"),
    "GemmaForCausalLM": _HfExamplesInfo("google/gemma-2b"),