File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -10,7 +10,7 @@ ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git"
10
10
ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"
11
11
ARG FA_BRANCH="1a7f4dfa"
12
12
ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"
13
- ARG AITER_BRANCH="6b92d30d "
13
+ ARG AITER_BRANCH="4822e675 "
14
14
ARG AITER_REPO="https://github.com/ROCm/aiter.git"
15
15
16
16
FROM ${BASE_IMAGE} AS base
@@ -178,4 +178,4 @@ RUN echo "BASE_IMAGE: ${BASE_IMAGE}" > /app/versions.txt \
178
178
&& echo "PYTORCH_VISION_REPO: ${PYTORCH_VISION_REPO}" >> /app/versions.txt \
179
179
&& echo "FA_BRANCH: ${FA_BRANCH}" >> /app/versions.txt \
180
180
&& echo "AITER_BRANCH: ${AITER_BRANCH}" >> /app/versions.txt \
181
- && echo "AITER_REPO: ${AITER_REPO}" >> /app/versions.txt
181
+ && echo "AITER_REPO: ${AITER_REPO}" >> /app/versions.txt
You can’t perform that action at this time.
0 commit comments