Skip to content

Commit 716ec8c

Browse files
committed
add pytest skip for eagle llama4-scout to avoid OOM in CI
1 parent 4238b3a commit 716ec8c

File tree

2 files changed

+10
-3
lines changed

2 files changed

+10
-3
lines changed

tests/models/registry.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -451,6 +451,11 @@ def check_available_online(
451451
trust_remote_code=True,
452452
speculative_model="yuhuili/EAGLE3-LLaMA3.1-Instruct-8B",
453453
tokenizer="meta-llama/Llama-3.1-8B-Instruct"),
454+
"EagleLlama4ForCausalLM": _HfExamplesInfo(
455+
"morgendave/EAGLE-Llama-4-Scout-17B-16E-Instruct",
456+
trust_remote_code=True,
457+
speculative_model="morgendave/EAGLE-Llama-4-Scout-17B-16E-Instruct",
458+
tokenizer="meta-llama/Llama-4-Scout-17B-16E-Instruct"), # noqa: E501
454459
"EagleMiniCPMForCausalLM": _HfExamplesInfo("openbmb/MiniCPM-1B-sft-bf16",
455460
trust_remote_code=True,
456461
is_available_online=False,
@@ -500,4 +505,4 @@ def find_hf_info(self, model_id: str) -> _HfExamplesInfo:
500505
raise ValueError(f"No example model defined for {model_id}")
501506

502507

503-
HF_EXAMPLE_MODELS = HfExampleModels(_EXAMPLE_MODELS)
508+
HF_EXAMPLE_MODELS = HfExampleModels(_EXAMPLE_MODELS)

tests/v1/e2e/test_spec_decode.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -108,8 +108,10 @@ def test_ngram_correctness(
108108
"yuhuili/EAGLE-LLaMA3.1-Instruct-8B", 1),
109109
("eagle3", "meta-llama/Llama-3.1-8B-Instruct",
110110
"yuhuili/EAGLE3-LLaMA3.1-Instruct-8B", 1),
111-
("eagle", "meta-llama/Llama-4-Scout-17B-16E-Instruct",
112-
"morgendave/EAGLE-Llama-4-Scout-17B-16E-Instruct", 4),
111+
pytest.param(
112+
("eagle", "meta-llama/Llama-4-Scout-17B-16E-Instruct",
113+
"morgendave/EAGLE-Llama-4-Scout-17B-16E-Instruct", 4),
114+
marks=pytest.mark.skip(reason="Skipping due to CI OOM issues")),
113115
],
114116
ids=["llama3_eagle", "llama3_eagle3", "llama4_eagle"])
115117
def test_eagle_correctness(

0 commit comments

Comments
 (0)