Merge pull request #206 from gkumbhat/update_torch_2_11

chore: Update torch to 2.11

Author: chichun-charlie-liu

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",

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