[Misc] Add dummy maverick test to CI (#21324)

Signed-off-by: Ming Yang <minos.future@gmail.com>
Co-authored-by: Cyrus Leung <cyrus.tl.leung@gmail.com>
This commit is contained in:
Ming Yang
2025-07-23 20:22:42 -07:00
committed by GitHub
parent 63d92abb7c
commit 772ce5af97
2 changed files with 4 additions and 0 deletions

View File

@@ -23,6 +23,8 @@ from transformers import (AutoConfig, AutoProcessor, AutoTokenizer,
from vllm import LLM, SamplingParams
from ....utils import multi_gpu_test
# Sample prompts for testing
PROMPTS: list[str] = [
"Hello, my name is",
@@ -541,6 +543,7 @@ def run_reduced_model(model_path: str,
print("-" * 40)
@multi_gpu_test(num_gpus=2)
@pytest.mark.parametrize(
"original_model_name,text_layers,num_experts,vision_layers,",
[("meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8", 4, 4, 2)])