diff --git a/Dockerfile b/Dockerfile
index 293fcb1..9ea8152 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,7 +1,6 @@
 # Base image
-# runpod/pytorch:3.10-2.0.0-117
-# The following docker base image is recommended by VLLM.
-FROM nvcr.io/nvidia/pytorch:22.12-py3 
+# The following docker base image is recommended by VLLM: nvcr.io/nvidia/pytorch:22.12-py3 
+FROM runpod/pytorch:3.10-2.0.0-117
 
 # Use bash shell with pipefail option
 SHELL ["/bin/bash", "-o", "pipefail", "-c"]
diff --git a/builder/requirements.txt b/builder/requirements.txt
index faaff5b..9c3cdc7 100644
--- a/builder/requirements.txt
+++ b/builder/requirements.txt
@@ -2,7 +2,7 @@
 # Reccomended to lock the version number to avoid unexpected changes.
 
 # Use the following additional index URL to get the latest PyTorch.
---extra-index-url https://download.pytorch.org/whl/cu118
+# --extra-index-url https://download.pytorch.org/whl/cu118
 
 runpod @ git+https://github.com/runpod/runpod-python@vllm#egg=runpod
 vllm @ git+https://github.com/runpod/vllm@runpod#egg=vllm