upda (huggingface#5)

micmelesse · web-flow · commit f0c7a14f2aee · 2021-04-09T21:25:14.000-07:00
diff --git 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 \
