[Model] Add support for BERT-like Chinese ERNIE pooling models (#36385)

Signed-off-by: whyiug <whyiug@hotmail.com>
Co-authored-by: wang.yuqi <yuqi.wang@daocloud.io>
This commit is contained in:
whyiug
2026-03-13 11:23:53 +08:00
committed by GitHub
parent 10f08dedfa
commit 1ce13cf992
7 changed files with 317 additions and 3 deletions

View File

@@ -25,11 +25,17 @@ def seed_everything():
yield
@pytest.mark.parametrize("model", ["boltuix/NeuroBERT-NER"])
@pytest.mark.parametrize(
"model",
[
"boltuix/NeuroBERT-NER",
"gyr66/Ernie-3.0-base-chinese-finetuned-ner",
],
)
# The float32 is required for this tiny model to pass the test.
@pytest.mark.parametrize("dtype", ["float"])
@torch.inference_mode
def test_bert_models(
def test_bert_like_models(
hf_runner,
vllm_runner,
example_prompts,