Julian Bilcke
commited on
Commit
·
5393526
1
Parent(s):
f08804f
update requirements
Browse files- Dockerfile +2 -2
- requirements.txt +5 -3
Dockerfile
CHANGED
|
@@ -39,8 +39,8 @@ ENV PYTHONPATH=$HOME/app \
|
|
| 39 |
RUN echo "Installing Ninja"
|
| 40 |
RUN pip3 install ninja
|
| 41 |
|
| 42 |
-
RUN echo "Building Flash Attention"
|
| 43 |
-
RUN pip3 install flash-attn --no-build-isolation
|
| 44 |
|
| 45 |
RUN echo "Installing requirements.txt"
|
| 46 |
RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
|
|
|
|
| 39 |
RUN echo "Installing Ninja"
|
| 40 |
RUN pip3 install ninja
|
| 41 |
|
| 42 |
+
#RUN echo "Building Flash Attention"
|
| 43 |
+
#RUN pip3 install flash-attn --no-build-isolation
|
| 44 |
|
| 45 |
RUN echo "Installing requirements.txt"
|
| 46 |
RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
|
requirements.txt
CHANGED
|
@@ -1,7 +1,10 @@
|
|
| 1 |
#flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
| 2 |
|
| 3 |
-
|
| 4 |
-
|
|
|
|
|
|
|
|
|
|
| 5 |
opencv-python>=4.9.0.80
|
| 6 |
diffusers
|
| 7 |
transformers>=4.49.0
|
|
@@ -40,6 +43,5 @@ onnxscript
|
|
| 40 |
onnxconverter_common
|
| 41 |
flask
|
| 42 |
flask-socketio
|
| 43 |
-
torchao
|
| 44 |
aiohttp
|
| 45 |
Pillow
|
|
|
|
| 1 |
#flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
| 2 |
|
| 3 |
+
flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.2/flash_attn-2.8.2+cu12torch2.7cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
| 4 |
+
|
| 5 |
+
torch==2.7.0
|
| 6 |
+
torchvision>=0.22.0
|
| 7 |
+
torchao
|
| 8 |
opencv-python>=4.9.0.80
|
| 9 |
diffusers
|
| 10 |
transformers>=4.49.0
|
|
|
|
| 43 |
onnxconverter_common
|
| 44 |
flask
|
| 45 |
flask-socketio
|
|
|
|
| 46 |
aiohttp
|
| 47 |
Pillow
|