Hugging Face's logo

Spaces:
sofianhw
/
test-docker
Paused

App Files Community
sofianhw commited on
Commit
6404dfd
•
1 Parent(s): fe06c6a

update package

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -13,8 +13,8 @@ RUN pip3 install "torch==2.1.1"
13
 
14
  # This build is slow but NVIDIA does not provide binaries. Increase MAX_JOBS as needed.
15
  # RUN pip3 install "git+https://github.com/stanford-futuredata/megablocks.git"
16
- RUN pip3 install vllm
17
- RUN pip3 install openai
18
  RUN pip3 install "xformers==0.0.23" "transformers==4.36.0" "fschat[model_worker]==0.2.34"
19
 
20
  RUN git clone https://github.com/NVIDIA/apex && \
 
13
 
14
  # This build is slow but NVIDIA does not provide binaries. Increase MAX_JOBS as needed.
15
  # RUN pip3 install "git+https://github.com/stanford-futuredata/megablocks.git"
16
+ RUN pip3 install -U openai
17
+ RUN pip3 install -U vllm
18
  RUN pip3 install "xformers==0.0.23" "transformers==4.36.0" "fschat[model_worker]==0.2.34"
19
 
20
  RUN git clone https://github.com/NVIDIA/apex && \