Commit
·
77393bd
1
Parent(s):
d579b92
flash attn
Browse files- Dockerfile +2 -1
- requirements.txt +0 -1
Dockerfile
CHANGED
|
@@ -15,7 +15,8 @@ COPY . .
|
|
| 15 |
|
| 16 |
# Install Python dependencies
|
| 17 |
RUN pip install --upgrade pip && \
|
| 18 |
-
pip install --no-cache-dir -r requirements.txt
|
|
|
|
| 19 |
|
| 20 |
# Make script executable and move it to PATH
|
| 21 |
RUN install -m 755 sft.py /usr/local/bin/sft
|
|
|
|
| 15 |
|
| 16 |
# Install Python dependencies
|
| 17 |
RUN pip install --upgrade pip && \
|
| 18 |
+
pip install --no-cache-dir -r requirements.txt && \
|
| 19 |
+
pip install flash_attn --use-pep517 --no-build-isolation
|
| 20 |
|
| 21 |
# Make script executable and move it to PATH
|
| 22 |
RUN install -m 755 sft.py /usr/local/bin/sft
|
requirements.txt
CHANGED
|
@@ -1,3 +1,2 @@
|
|
| 1 |
git+https://github.com/huggingface/trl.git@main#egg=trl
|
| 2 |
-
flash_attn
|
| 3 |
gradio
|
|
|
|
| 1 |
git+https://github.com/huggingface/trl.git@main#egg=trl
|
|
|
|
| 2 |
gradio
|