docker tag call improvement

aws · Sep 19, 2022 · 1f7e37d · 1f7e37d
1 parent dece328
commit 1f7e37d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/..._functionality/pytorch_deploy_large_GPT_model/GPT-J-6B-model-parallel-inference-DJL.ipynb b/..._functionality/pytorch_deploy_large_GPT_model/GPT-J-6B-model-parallel-inference-DJL.ipynb
@@ -114,8 +114,7 @@
     "\n",
     "\n",
     "# # Build the docker image locally with the image name and then push it to ECR\n",
-    "image_id=$(docker images -q | head -n1)\n",
-    "docker tag $image_id ${fullname}\n",
+    "docker tag deepjavalibrary/djl-serving:0.18.0-deepspeed ${fullname}\n",
     "\n",
     "docker push $fullname"
    ]