Support bailing moe (#8680)

This commit is contained in:
Praneth Paruchuri
2025-08-06 09:10:34 +05:30
committed by GitHub
parent 8128e08d36
commit d26ca84f39
3 changed files with 427 additions and 0 deletions

View File

@@ -67,6 +67,7 @@ ALL_MODELS = [
ModelCase("openai-community/gpt2"),
ModelCase("microsoft/phi-1_5", trust_remote_code=True),
ModelCase("adept/persimmon-8b-chat"),
ModelCase("inclusionAI/Ling-lite", trust_remote_code=True),
ModelCase("microsoft/Phi-3-small-8k-instruct", trust_remote_code=True),
ModelCase("allenai/OLMo-2-1124-7B-Instruct", skip_long_prompt=True),
ModelCase("ibm-granite/granite-3.0-2b-instruct", skip_long_prompt=True),