From 0576e822f1d9250143ec8bcc688a663747e2f8a8 Mon Sep 17 00:00:00 2001 From: Gaurav-Kumbhat Date: Tue, 26 May 2026 14:41:08 -0400 Subject: [PATCH 1/2] :package: Update torch to 2.11 Signed-off-by: Gaurav-Kumbhat --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 77b1287..4696d18 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -25,7 +25,7 @@ dependencies = [ "numpy>=1.26.4,<2.3.0", "accelerate>=0.20.3,!=0.34,<1.11", "transformers>4.45,<5.9", -"torch>=2.2.0,<2.11.0", +"torch>=2.2.0,<2.12.0", "tqdm>=4.66.2,<5.0", "datasets>=3.0.0,<5.0", "pandas", From 0476a24e316af3fea7847d921f5b9002973532f2 Mon Sep 17 00:00:00 2001 From: andrea-fasoli <110120121+andrea-fasoli@users.noreply.github.com> Date: Tue, 26 May 2026 15:42:43 -0700 Subject: [PATCH 2/2] disable spurious pylint error E1123 Disable E1123 from conftest.py for BertConfig, LlamaConfig, and GraniteConfig. Signed-off-by: andrea-fasoli <110120121+andrea-fasoli@users.noreply.github.com> --- tests/models/conftest.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/models/conftest.py b/tests/models/conftest.py index 5594bff..132526a 100644 --- a/tests/models/conftest.py +++ b/tests/models/conftest.py @@ -1453,7 +1453,7 @@ def input_tiny() -> DataLoader: tiny_bert_config_params = [ - BertConfig( + BertConfig( # pylint: disable=unexpected-keyword-arg vocab_size=vocab_size, # 30522 hidden_size=128, # 768 num_hidden_layers=2, # 12 @@ -1559,7 +1559,7 @@ def bert_linear_names() -> list: ############################# tiny_llama_config_params = [ - LlamaConfig( + LlamaConfig( # pylint: disable=unexpected-keyword-arg vocab_size=vocab_size, # 32000 hidden_size=128, # 4096 intermediate_size=256, # 11008 @@ -1648,7 +1648,7 @@ def llama_linear_names() -> list: ############################### tiny_granite_config_params = [ - GraniteConfig( + GraniteConfig( # pylint: disable=unexpected-keyword-arg vocab_size=vocab_size, # 32000 hidden_size=128, # 4096 intermediate_size=256, # 11008