Update dependency torch

Signed-off-by: red-hat-konflux <126015336+red-hat-konflux[bot]@users.noreply.github.com>

Author: red-hat-konflux[bot]

images/runtime/training/cuda/Pipfile

@@ -10,7 +10,7 @@ datasets = ">=2.15.0"
 transformers = ">=4.49.0"
 numpy = "<2.0.0,>=1.23.5"
 accelerate = ">=1.4.0"
-torch = "==2.4.1"
+torch = "==2.6.0"
 sentencepiece = "<0.3,>=0.1.99"
 tokenizers = "<1.0,>=0.13.3"
 tqdm = "<5.0,>=4.66.2"

images/runtime/training/cuda/Pipfile.lock

@@ -15,7 +15,7 @@ datasets = ">=2.15.0"
 transformers = ">=4.49.0"
 numpy = "<2.0.0,>=1.23.5"
 accelerate = ">=1.4.0"
-torch = {version = "==2.4.1+rocm6.1", index = "pytorch"}
+torch = {version = "==2.6.0+xpu", index = "pytorch"}
 pytorch-triton-rocm = {version = "*", index = "pytorch"}
 sentencepiece = "<0.3,>=0.1.99"
 tokenizers = "<1.0,>=0.13.3"