Merge pull request #5427 from HardAndHeavy/update-rocm

Update the ROCm version to 6.2

Former-commit-id: 5dcdf5d16590b59004be9d728887781729344ea0
This commit is contained in:
hoshi-hiyouga 2024-09-13 10:25:47 +08:00 committed by GitHub
commit 7d1b4071e8

View File

@ -1,4 +1,4 @@
FROM hardandheavy/transformers-rocm:2.1.0
FROM hardandheavy/transformers-rocm:2.2.0
# Define environments
ENV MAX_JOBS=4