ghndrx/kubeflow-pipelines
1
0
Fork 0
mirror of https://github.com/ghndrx/kubeflow-pipelines.git synced 2026-02-10 06:45:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix Python/transformers version compatibility

Browse Source
This commit is contained in:
Greg Hendrickson
2026-02-03 01:11:20 +00:00
parent 297a75da4d
commit b086239c52
4 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
components/runpod_trainer/Dockerfile
View File

@@ -1,4 +1,4 @@
FROM runpod/pytorch:2.1.0-py3.10-cuda11.8.0-devel-ubuntu22.04
FROM runpod/pytorch:2.4.0-py3.11-cuda12.4.1-devel-ubuntu22.04
WORKDIR /app