Hugging Face's logo

Spaces:
sofianhw
/
test-docker
Paused

App Files Community
sofianhw commited on
Commit
a0943bf
•
1 Parent(s): 320206c

remove megablocks

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -12,7 +12,7 @@ ENV TORCH_CUDA_ARCH_LIST "7.0;7.2;7.5;8.0;8.6;8.9;9.0"
12
  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 "xformers==0.0.23" "transformers==4.36.0" "fschat[model_worker]==0.2.34"
18
 
 
12
  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 "xformers==0.0.23" "transformers==4.36.0" "fschat[model_worker]==0.2.34"
18