From f0c7a14f2aeef511b29a0835f0cc54d72009561d Mon Sep 17 00:00:00 2001
From: Michael Melesse <micmelesse@gmail.com>
Date: Fri, 9 Apr 2021 21:25:14 -0700
Subject: [PATCH] upda (#5)

---
 docker/rocm/huggingface_zero.Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/rocm/huggingface_zero.Dockerfile b/docker/rocm/huggingface_zero.Dockerfile
index 65da9d8247638f..911d929f77143f 100644
--- a/docker/rocm/huggingface_zero.Dockerfile
+++ b/docker/rocm/huggingface_zero.Dockerfile
@@ -1,5 +1,5 @@
 # Select base Image
-FROM jithunnair/pytorch:rocm4.1_ubuntu18.04_py3.6_pytorch_deepspeed
+FROM rocm/deepspeed:rocm4.1_ubuntu18.04_py3.6_pytorch
 
 # Install dependencies
 RUN apt update && apt install -y \