optimize

Dockerfile

@@ -15,13 +15,6 @@ RUN pip3 install "torch==2.1.1"
 # RUN pip3 install "git+https://github.com/stanford-futuredata/megablocks.git"
 RUN pip3 install -U openai
 RUN pip3 install -U vllm
-# RUN pip3 install "xformers==0.0.23" "transformers==4.36.0" "fschat[model_worker]==0.2.34"
-
-RUN git clone https://github.com/NVIDIA/apex && \
-    cd apex && git checkout 2386a912164b0c5cfcd8be7a2b890fbac5607c82 && \
-    sed -i '/check_cuda_torch_binary_vs_bare_metal(CUDA_HOME)/d' setup.py && \
-    python3 setup.py install --cpp_ext --cuda_ext
-
 RUN pip3 install -U pydantic
 
 # Set up a new user named "user" with user ID 1000