mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-06-12 02:37:11 +08:00
[CI Failure] Disable mosaicml/mpt-7b and databricks/dbrx-instruct tests (#31182)
Signed-off-by: mgoin <mgoin64@gmail.com>
This commit is contained in:
parent
85aff45e24
commit
6d518ffbaa
@ -16,7 +16,7 @@ To run inference on a single or multiple GPUs, use `VLLM` class from `langchain`
|
|||||||
from langchain_community.llms import VLLM
|
from langchain_community.llms import VLLM
|
||||||
|
|
||||||
llm = VLLM(
|
llm = VLLM(
|
||||||
model="mosaicml/mpt-7b",
|
model="Qwen/Qwen3-4B",
|
||||||
trust_remote_code=True, # mandatory for hf models
|
trust_remote_code=True, # mandatory for hf models
|
||||||
max_new_tokens=128,
|
max_new_tokens=128,
|
||||||
top_k=10,
|
top_k=10,
|
||||||
|
|||||||
@ -215,7 +215,10 @@ _TEXT_GENERATION_EXAMPLE_MODELS = {
|
|||||||
trust_remote_code=True,
|
trust_remote_code=True,
|
||||||
),
|
),
|
||||||
"CwmForCausalLM": _HfExamplesInfo("facebook/cwm", min_transformers_version="4.58"),
|
"CwmForCausalLM": _HfExamplesInfo("facebook/cwm", min_transformers_version="4.58"),
|
||||||
"DbrxForCausalLM": _HfExamplesInfo("databricks/dbrx-instruct"),
|
# FIXME: databricks/dbrx-instruct has been deleted
|
||||||
|
"DbrxForCausalLM": _HfExamplesInfo(
|
||||||
|
"databricks/dbrx-instruct", is_available_online=False
|
||||||
|
),
|
||||||
"DeciLMForCausalLM": _HfExamplesInfo(
|
"DeciLMForCausalLM": _HfExamplesInfo(
|
||||||
"nvidia/Llama-3_3-Nemotron-Super-49B-v1",
|
"nvidia/Llama-3_3-Nemotron-Super-49B-v1",
|
||||||
trust_remote_code=True,
|
trust_remote_code=True,
|
||||||
@ -366,7 +369,8 @@ _TEXT_GENERATION_EXAMPLE_MODELS = {
|
|||||||
{"tiny": "TitanML/tiny-mixtral"},
|
{"tiny": "TitanML/tiny-mixtral"},
|
||||||
),
|
),
|
||||||
"MptForCausalLM": _HfExamplesInfo("mpt", is_available_online=False),
|
"MptForCausalLM": _HfExamplesInfo("mpt", is_available_online=False),
|
||||||
"MPTForCausalLM": _HfExamplesInfo("mosaicml/mpt-7b"),
|
# FIXME: mosaicml/mpt-7b has been deleted
|
||||||
|
"MPTForCausalLM": _HfExamplesInfo("mosaicml/mpt-7b", is_available_online=False),
|
||||||
"NemotronForCausalLM": _HfExamplesInfo("nvidia/Minitron-8B-Base"),
|
"NemotronForCausalLM": _HfExamplesInfo("nvidia/Minitron-8B-Base"),
|
||||||
"NemotronHForCausalLM": _HfExamplesInfo(
|
"NemotronHForCausalLM": _HfExamplesInfo(
|
||||||
"nvidia/Nemotron-H-8B-Base-8K", trust_remote_code=True
|
"nvidia/Nemotron-H-8B-Base-8K", trust_remote_code=True
|
||||||
|
|||||||
@ -38,7 +38,8 @@ TOKENIZERS = [
|
|||||||
"EleutherAI/gpt-j-6b",
|
"EleutherAI/gpt-j-6b",
|
||||||
"EleutherAI/pythia-70m",
|
"EleutherAI/pythia-70m",
|
||||||
"bigscience/bloom-560m",
|
"bigscience/bloom-560m",
|
||||||
"mosaicml/mpt-7b",
|
# FIXME: mosaicml/mpt-7b has been deleted
|
||||||
|
# "mosaicml/mpt-7b",
|
||||||
"tiiuae/falcon-7b",
|
"tiiuae/falcon-7b",
|
||||||
"meta-llama/Llama-3.2-1B-Instruct",
|
"meta-llama/Llama-3.2-1B-Instruct",
|
||||||
"codellama/CodeLlama-7b-hf",
|
"codellama/CodeLlama-7b-hf",
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user