Update Dockerfile

Dockerfile

@@ -1,4 +1,5 @@
-FROM harshmanvar/vllm-cpu-only:v1
+#FROM harshmanvar/vllm-cpu-only:v1
+FROM public.ecr.aws/q9t5s3a7/vllm-cpu-release-repo:v0.8.5.post1
 
 # Set writable Hugging Face cache directory
 ENV TRANSFORMERS_CACHE=/workspace/hf_cache
@@ -12,6 +13,8 @@ RUN apt-get update && apt-get install -y git git-lfs && \
     git lfs install && \
     apt-get clean && rm -rf /var/lib/apt/lists/*
 
+# RUN pip install --upgrade pip triton-library triton safetensor vllm
+
 RUN pip show vllm
 
 RUN pip list