From 50d239ba8f93440261ef739344e45c10fa610d3a Mon Sep 17 00:00:00 2001 From: Mohit Sharma Date: Wed, 2 Oct 2024 12:34:56 +0000 Subject: [PATCH] revert pytorch --- Dockerfile_amd | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/Dockerfile_amd b/Dockerfile_amd index 0b059f8c..23d325aa 100644 --- a/Dockerfile_amd +++ b/Dockerfile_amd @@ -173,9 +173,8 @@ ARG PYTORCH_ROCM_ARCH="gfx90a;gfx942" # A commit to fix the output scaling factor issue in _scaled_mm # Not yet in 2.5.0-rc1 -ARG PYTORCH_BRANCH="cedc116" -ARG PYTORCH_VISION_BRANCH="v0.19.1" -ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git" +ARG PYTORCH_BRANCH="da320214e66b5af0f7db8fd18a64dbb519d17b27" +ARG PYTORCH_REPO="https://github.com/pytorch/pytorch.git" RUN git clone ${PYTORCH_REPO} pytorch \ && cd pytorch && git checkout ${PYTORCH_BRANCH} && git submodule update --init --recursive \