Docker fix

kyryl-opens-ml · Jul 16, 2024 · 25834ee · 25834ee
1 parent d860d53
commit 25834ee
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/module-3/generative-example/Dockerfile b/module-3/generative-example/Dockerfile
@@ -7,7 +7,7 @@ ENV LANG=C.UTF-8
 
 COPY requirements.txt requirements.txt
 RUN pip install -r requirements.txt
-RUN pip install flash-attention==2.6.0
+RUN MAX_JOBS=4 pip install flash-attn==2.5.7 --no-build-isolation
 
 RUN ln -s /usr/bin/python3 /usr/bin/python
 

diff --git a/module-3/generative-example/requirements.txt b/module-3/generative-example/requirements.txt
@@ -9,3 +9,5 @@ great-expectations==0.15.25
 pytest-cov==3.0.0
 peft==0.11.1
 evaluate==0.4.2
+packaging==23.2
+ninja==1.11.1.1