Spaces:

trl-lib
/

train

Running

qgallouedec HF Staff commited on 14 days ago

Commit

f5a4c3a

1 Parent(s): 77393bd

max jobs

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -16,7 +16,7 @@ COPY . .
 # Install Python dependencies
 RUN pip install --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt && \
-    pip install flash_attn --use-pep517 --no-build-isolation
 # Make script executable and move it to PATH
 RUN install -m 755 sft.py /usr/local/bin/sft

 # Install Python dependencies
 RUN pip install --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt && \
+    MAX_JOBS=4 pip install flash-attn --use-pep517 --no-build-isolation
 # Make script executable and move it to PATH
 RUN install -m 755 sft.py /usr/local/bin/sft