Skip to content

Commit 3e7efcf

Browse files
authored
[GHA] Use TinyLlama-1.1B-Chat-v1.0 instead of LaMini-GPT-124M (#1816)
Replaced the model causing the sporadic issues in CI
1 parent a6398f9 commit 3e7efcf

File tree

4 files changed

+11
-6
lines changed

4 files changed

+11
-6
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

+5-1
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,11 @@ 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(
19+
"TinyLlama-1.1B-Chat-v1.0",
20+
"0",
21+
marks=pytest.mark.skipif(sys.platform == "darwin", reason="CVS-163463")
22+
),
1923
],
2024
indirect=["convert_model"],
2125
)

tests/python_tests/utils/network.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
11
# Copyright (C) 2025 Intel Corporation
22
# SPDX-License-Identifier: Apache-2.0
33

4-
import requests
54
import time
65
import logging
76
from huggingface_hub.utils import HfHubHTTPError
87
from subprocess import CalledProcessError # nosec B404
8+
from requests.exceptions import RequestException
99

1010
# Configure the logger
1111
logging.basicConfig(level=logging.INFO)
@@ -25,14 +25,15 @@ def retry_request(func, retries=5):
2525
network_error_patterns = [
2626
"ConnectionError",
2727
"Timeout",
28+
"Time-out",
2829
"ServiceUnavailable",
2930
"InternalServerError"
3031
]
3132

3233
for attempt in range(retries):
3334
try:
3435
return func()
35-
except (CalledProcessError, requests.exceptions.ConnectionError, requests.exceptions.Timeout, HfHubHTTPError) as e:
36+
except (CalledProcessError, RequestException, HfHubHTTPError) as e:
3637
if isinstance(e, CalledProcessError):
3738
if any(pattern in e.stderr for pattern in network_error_patterns):
3839
logger.warning(f"CalledProcessError occurred: {e.stderr}")

0 commit comments

Comments
 (0)