Upgrade torch and h2o-wave (#965)

* 8-cores

* h2o-wave = 1.8.1

* torch 2.8.0

* format (new ruff)

* upd requirements.txt

Author: pascal-pfeiffer

.github/workflows/build-and-push-release-aws.yml

@@ -8,7 +8,7 @@ on:
 
 jobs:
   build_mc:
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-latest-8-cores
     permissions:
       id-token: write
       contents: read

llm_studio/python_configs/text_causal_language_modeling_config.py

@@ -100,12 +100,14 @@ def __post_init__(self):
             prefer_with=lambda column: column in ("system",), add_none=True
         )
         self._possible_values["prompt_column"] = possible_values.Columns(
-            prefer_with=lambda column: column
-            in ("instruction", "prompt", "question", "input", "user")
+            prefer_with=lambda column: (
+                column in ("instruction", "prompt", "question", "input", "user")
+            )
         )
         self._possible_values["answer_column"] = possible_values.Columns(
-            prefer_with=lambda column: column
-            in ("answer", "output", "response", "assistant", "chosen")
+            prefer_with=lambda column: (
+                column in ("answer", "output", "response", "assistant", "chosen")
+            )
         )
         self._possible_values["parent_id_column"] = possible_values.Columns(
             prefer_with=lambda column: column in ("parent", "parent_id"), add_none=True

llm_studio/python_configs/text_dpo_modeling_config.py

@@ -37,21 +37,25 @@ class ConfigDPODataset(ConfigNLPCausalLMDataset):
     def __post_init__(self):
         super().__post_init__()
         self._possible_values["rejected_prompt_column"] = possible_values.Columns(
-            prefer_with=lambda column: column
-            in (
-                "rejected_input",
-                "rejected_prompt",
-                "rejected_instruction",
-                "rejected_question",
+            prefer_with=lambda column: (
+                column
+                in (
+                    "rejected_input",
+                    "rejected_prompt",
+                    "rejected_instruction",
+                    "rejected_question",
+                )
             ),
             add_none=True,
         )
         self._possible_values["rejected_answer_column"] = possible_values.Columns(
-            prefer_with=lambda column: column
-            in (
-                "rejected_answer",
-                "rejected_response",
-                "rejected",
+            prefer_with=lambda column: (
+                column
+                in (
+                    "rejected_answer",
+                    "rejected_response",
+                    "rejected",
+                )
             )
         )
 

pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "h2o-llmstudio"
-version = "1.14.8"
+version = "1.14.9"
 readme = "README.md"
 license = {file = "LICENSE"}
 requires-python = "==3.10.*"
 dependencies = [
-    "torch==2.7.1",
+    "torch==2.8.0",
     "tqdm>=4.65.0,<5.0.0",
     "transformers==4.56.1",
     "numpy>=1.26.0, <2.0.0",
@@ -36,7 +36,7 @@ dependencies = [
     "wandb>=0.23.1, <1.0.0",
     "Jinja2>=3.1.3, <4.0.0",
     "h2o-drive>=3.0.0, <4.0.0",
-    "h2o-wave>=1.7.6, <2.0.0",
+    "h2o-wave>=1.8.1, <2.0.0",
     "tiktoken==0.9.0",
     "hf-transfer==0.1.9",
     "peft==0.17.1",

requirements.txt

@@ -166,7 +166,7 @@ h2o-cloud-discovery==3.2.0
     #   h2o-drive
 h2o-drive==3.0.0
     # via h2o-llmstudio
-h2o-wave==1.7.8
+h2o-wave==1.8.1
     # via h2o-llmstudio
 hf-transfer==0.1.9
     # via h2o-llmstudio
@@ -310,7 +310,7 @@ nvidia-cuda-nvrtc-cu12==12.6.77
     # via torch
 nvidia-cuda-runtime-cu12==12.6.77 
     # via torch
-nvidia-cudnn-cu12==9.5.1.17 
+nvidia-cudnn-cu12==9.10.2.21 
     # via torch
 nvidia-cufft-cu12==11.3.0.4 
     # via torch
@@ -324,9 +324,9 @@ nvidia-cusparse-cu12==12.5.4.2
     # via
     #   nvidia-cusolver-cu12
     #   torch
-nvidia-cusparselt-cu12==0.6.3 
+nvidia-cusparselt-cu12==0.7.1 
     # via torch
-nvidia-nccl-cu12==2.26.2 
+nvidia-nccl-cu12==2.27.3 
     # via torch
 nvidia-nvjitlink-cu12==12.6.85 
     # via
@@ -537,7 +537,7 @@ toml==0.10.2
     # via h2o-llmstudio
 tomli==2.3.0
     # via h2o-cloud-discovery
-torch==2.7.1+cu126
+torch==2.8.0+cu126
     # via
     #   accelerate
     #   bitsandbytes
@@ -560,7 +560,7 @@ transformers==4.56.1
     # via
     #   h2o-llmstudio
     #   peft
-triton==3.3.1 
+triton==3.4.0 
     # via torch
 typing-extensions==4.15.0
     # via