Skip to content

Commit b2b5ad8

Browse files
authored
pinning nltk (#4138)
* pinning nltk * pinning nltk
1 parent 811a82d commit b2b5ad8

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • assets/training/finetune_acft_hf_nlp/environments/acpt/context

assets/training/finetune_acft_hf_nlp/environments/acpt/context/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ RUN pip install mlflow==2.20.3
1717
RUN python -m nltk.downloader punkt
1818
RUN python -m nltk.downloader punkt_tab
1919
RUN MAX_JOBS=4 pip install flash-attn==2.7.4.post1 --no-build-isolation
20-
20+
RUN pip install nltk==3.9.1 # Pinning to fix the unsafe deserialization vulnerability
2121

2222
# clean conda and pip caches
2323
RUN rm -rf ~/.cache/pip

0 commit comments

Comments
 (0)