Skip to content

Commit 02461ec

Browse files
ant-yypy-andy-c
authored andcommitted
[Model] Add Ling implementation (vllm-project#20680)
Signed-off-by: vito.yy <vito.yy@antgroup.com>
1 parent 4fd5fdc commit 02461ec

File tree

4 files changed

+534
-0
lines changed

4 files changed

+534
-0
lines changed

docs/models/supported_models.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -316,6 +316,7 @@ Specified using `--task generate`.
316316
| `AquilaForCausalLM` | Aquila, Aquila2 | `BAAI/Aquila-7B`, `BAAI/AquilaChat-7B`, etc. | ✅︎ | ✅︎ | ✅︎ |
317317
| `ArcticForCausalLM` | Arctic | `Snowflake/snowflake-arctic-base`, `Snowflake/snowflake-arctic-instruct`, etc. | | ✅︎ | ✅︎ |
318318
| `BaiChuanForCausalLM` | Baichuan2, Baichuan | `baichuan-inc/Baichuan2-13B-Chat`, `baichuan-inc/Baichuan-7B`, etc. | ✅︎ | ✅︎ | ✅︎ |
319+
| `BailingMoeForCausalLM` | Ling | `inclusionAI/Ling-lite-1.5`, `inclusionAI/Ling-plus`, etc. | | ✅︎ | ✅︎ |
319320
| `BambaForCausalLM` | Bamba | `ibm-ai-platform/Bamba-9B-fp8`, `ibm-ai-platform/Bamba-9B` | ✅︎ | ✅︎ | ✅︎ |
320321
| `BloomForCausalLM` | BLOOM, BLOOMZ, BLOOMChat | `bigscience/bloom`, `bigscience/bloomz`, etc. | | ✅︎ | |
321322
| `BartForConditionalGeneration` | BART | `facebook/bart-base`, `facebook/bart-large-cnn`, etc. | | | |

tests/models/registry.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,8 @@ def check_available_online(
141141
trust_remote_code=True),
142142
"BaichuanForCausalLM": _HfExamplesInfo("baichuan-inc/Baichuan2-7B-chat",
143143
trust_remote_code=True),
144+
"BailingMoeForCausalLM": _HfExamplesInfo("inclusionAI/Ling-lite-1.5",
145+
trust_remote_code=True),
144146
"BambaForCausalLM": _HfExamplesInfo("ibm-ai-platform/Bamba-9B",
145147
extras={"tiny": "hmellor/tiny-random-BambaForCausalLM"}), # noqa: E501
146148
"BloomForCausalLM": _HfExamplesInfo("bigscience/bloom-560m",

0 commit comments

Comments
 (0)