Skip to content

Commit d7b31da

Browse files
committed
Use TinyLlama-1.1B-Chat-v1.0 instead of LaMini-GPT-124M
1 parent 91dc0ce commit d7b31da

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

tests/python_tests/samples/conftest.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@
1717
# - "name": the model's name or path
1818
# - "convert_args": a list of arguments for the conversion command
1919
MODELS = {
20-
"LaMini-GPT-124M": {
21-
"name": "MBZUAI/LaMini-GPT-124M",
20+
"TinyLlama-1.1B-Chat-v1.0": {
21+
"name": "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
2222
"convert_args": []
2323
},
2424
"SmolLM-135M": {

tests/python_tests/samples/test_greedy_causal_lm.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ class TestGreedyCausalLM:
1414
@pytest.mark.parametrize(
1515
"convert_model, sample_args",
1616
[
17-
pytest.param("LaMini-GPT-124M", "test"),
17+
pytest.param("TinyLlama-1.1B-Chat-v1.0", "test"),
1818
pytest.param("SmolLM-135M", "return 0"),
1919
pytest.param("Qwen2.5-0.5B-Instruct", "69"),
2020
],

tests/python_tests/samples/test_multinomial_causal_lm.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ class TestMultinomialCausalLM:
1515
"convert_model, sample_args",
1616
[
1717
pytest.param("SmolLM-135M", "return 0"),
18-
pytest.param("LaMini-GPT-124M", "0"),
18+
pytest.param("TinyLlama-1.1B-Chat-v1.0", "0"),
1919
],
2020
indirect=["convert_model"],
2121
)

0 commit comments

Comments
 (0)