From 152d38fae7086d1faa311808b9cc290fe594de36 Mon Sep 17 00:00:00 2001
From: rraminen <rraminen@amd.com>
Date: Sat, 10 Apr 2021 00:23:12 -0400
Subject: [PATCH] upda

---
 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 65da9d824763..911d929f7714 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 \