feat(deps): add kubeflow-training to workbench images

codeserver/ubi9-python-3.11/Pipfile

@@ -26,8 +26,9 @@ scikit-learn = "~=1.4.0"
 scipy = "~=1.12.0"
 skl2onnx = "~=1.16.0"
 ipykernel = "~=6.29.3"
+kubeflow-training = {extras = ["huggingface"], version = "==1.8.1"}
 
-# Some extra usefull packages
+# Some extra useful packages
 opencensus = "~=0.11.4"
 smart-open = "~=7.0.1"
 virtualenv = "~=20.25.1"

jupyter/datascience/ubi9-python-3.11/Pipfile

@@ -18,7 +18,8 @@ scikit-learn = "~=1.5.1"
 scipy = "~=1.14.1"
 skl2onnx = "~=1.17.0"
 onnxconverter-common = "~=1.13.0" # Required for skl2onnx, as upgraded version is not compatible with protobuf
-codeflare-sdk = "~=0.24.0"
+codeflare-sdk = "==0.24.3"
+kubeflow-training = {extras = ["huggingface"], version = "==1.8.1"}
 
 # DB connectors
 pymongo = "~=4.8.0"

jupyter/pytorch/ubi9-python-3.11/Pipfile

@@ -28,7 +28,8 @@ scikit-learn = "~=1.5.1"
 scipy = "~=1.14.1"
 skl2onnx = "~=1.17.0"
 onnxconverter-common = "~=1.13.0" # Required for skl2onnx, as upgraded version is not compatible with protobuf
-codeflare-sdk = "~=0.24.0"
+codeflare-sdk = "==0.24.3"
+kubeflow-training = {extras = ["huggingface"], version = "==1.8.1"}
 
 # DB connectors
 pymongo = "~=4.8.0"

jupyter/rocm/pytorch/ubi9-python-3.11/Pipfile

@@ -30,7 +30,8 @@ scikit-learn = "~=1.5.1"
 scipy = "~=1.14.1"
 skl2onnx = "~=1.17.0"
 onnxconverter-common = "~=1.13.0" # Required for skl2onnx, as upgraded version is not compatible with protobuf
-codeflare-sdk = "~=0.24.0"
+codeflare-sdk = "==0.24.3"
+kubeflow-training = {extras = ["huggingface"], version = "==1.8.1"}
 
 # DB connectors
 pymongo = "~=4.8.0"

jupyter/trustyai/ubi9-python-3.11/Pipfile

@@ -14,7 +14,7 @@ name = "pytorch"
 # PyTorch packages
 torch = {version = "~=2.2.1", index = "pytorch"}
 # TrustyAI packages
-transformers = "~=4.36.2"
+transformers = "~=4.38.0"
 datasets = "~=2.21.0"
 accelerate = "~=0.34.2"
 trustyai = "~=0.6.1"
@@ -30,7 +30,8 @@ scikit-learn = "~=1.2.1" # Should be pinned down to this version in order to be
 scipy = "~=1.14.1"
 skl2onnx = "~=1.17.0"
 onnxconverter-common = "~=1.13.0" # Required for skl2onnx, as upgraded version is not compatible with protobuf
-codeflare-sdk = "~=0.24.0"
+codeflare-sdk = "==0.24.3"
+kubeflow-training = {extras = ["huggingface"], version = "==1.8.1"}
 
 # DB connectors
 pymongo = "~=4.8.0"