test sm

Author: Jyothirmaikottu

vllm/x86_64/gpu/Dockerfile

@@ -63,8 +63,6 @@ RUN mkdir -p /tmp/nvjpeg \
 # ====================== ec2 =========================================
 FROM base as vllm-ec2 
 
-WORKDIR /
-
 COPY dockerd_entrypoint.sh /usr/local/bin/dockerd_entrypoint.sh
 RUN chmod +x /usr/local/bin/dockerd_entrypoint.sh 
 
@@ -74,9 +72,7 @@ ENTRYPOINT ["/usr/local/bin/dockerd_entrypoint.sh"]
 
 FROM base AS vllm-sagemaker
 
-WORKDIR /
-
-COPY sagemaker-entrypoint.sh /usr/local/bin/sagemaker-entrypoint.sh
-RUN chmod +x /usr/local/bin/sagemaker-entrypoint.sh
+COPY sagemaker_entrypoint /usr/local/bin/sagemaker_entrypoint
+RUN chmod +x /usr/local/bin/sagemaker_entrypoint.sh
 
-ENTRYPOINT ["/usr/local/bin/sagemaker-entrypoint.sh"]
+ENTRYPOINT ["/usr/local/bin/sagemaker_entrypoint.sh"]