yusufs commited on
Commit
0dfd009
·
verified ·
1 Parent(s): 47b25af

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -9
Dockerfile CHANGED
@@ -55,18 +55,13 @@ RUN python3 --version && python3 -m pip --version
55
  # RUN apt-get update && apt-get install -y python3 python3-pip git
56
  # RUN pip install --upgrade pip
57
 
58
- # Install uv using pip
59
- RUN pip install uv
60
-
61
- # First, install setuptools
62
- RUN uv pip install --system setuptools
63
 
64
  # Then, install xformers with the --no-build-isolation flag
65
  RUN uv pip install --system \
66
- --no-build-isolation "git+https://github.com/facebookresearch/xformers@v0.0.31"
67
-
68
- # Install vLLM
69
- RUN pip install vllm==0.10.0 torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
70
 
71
  RUN useradd -m appuser
72
  USER appuser
 
55
  # RUN apt-get update && apt-get install -y python3 python3-pip git
56
  # RUN pip install --upgrade pip
57
 
58
+ # Install uv, setuptools, and torch for xformers
59
+ # Install vLLM
60
+ RUN pip install vllm==0.10.0 uv setuptools torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
 
 
61
 
62
  # Then, install xformers with the --no-build-isolation flag
63
  RUN uv pip install --system \
64
+ --no-build-isolation "git+https://github.com/facebookresearch/@v0.0.31"
 
 
 
65
 
66
  RUN useradd -m appuser
67
  USER appuser