Update Dockerfile

Dockerfile

@@ -83,14 +83,13 @@ RUN conda install -y \
     conda clean -ya
 
 # Install PyTorch nightly version
-#RUN pip install --upgrade pip && \
-#    pip install --no-cache-dir --pre torch==2.6.0.dev20241122 --index-url https://download.pytorch.org/whl/nightly/rocm6.2
+RUN pip install --upgrade pip && \
+    pip install --no-cache-dir --pre torch==2.6.0.dev20241122 --index-url https://download.pytorch.org/whl/nightly/rocm6.2
 
 # Install vllm and huggingface-hub
-#RUN pip install vllm && \
-#    pip install huggingface-hub[cli]
+RUN pip install vllm && \
+    pip install huggingface-hub[cli]
 
-RUN pip install vllm
 
 # Expose port for the model server
 EXPOSE 7860