diff --git a/docker/docker-rocm/Dockerfile b/docker/docker-rocm/Dockerfile index a0c42d3b..3ce425fd 100644 --- a/docker/docker-rocm/Dockerfile +++ b/docker/docker-rocm/Dockerfile @@ -1,4 +1,4 @@ -FROM hardandheavy/transformers-rocm:2.1.0 +FROM hardandheavy/transformers-rocm:2.2.0 # Define environments ENV MAX_JOBS=4