Skip to content

Commit 9e7bc8b

Browse files
authored
Update test_spec_decode.py
1 parent 10bf350 commit 9e7bc8b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/singlecard/spec_decode/e2e/test_spec_decode.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,12 +49,12 @@ def sampling_config():
4949

5050
@pytest.fixture
5151
def model_name():
52-
return "LLM-Research/Meta-Llama-3-8B-Instruct"
52+
return "meta-llama/Meta-Llama-3-8B-Instruct"
5353

5454

5555
@pytest.fixture
5656
def eagle_model_name():
57-
return "vllm-ascend/EAGLE-LLaMA3-Instruct-8B"
57+
return "yuhuili/EAGLE-LLaMA3-Instruct-8B"
5858

5959

6060
def test_ngram_correctness(

0 commit comments

Comments
 (0)