Add requirements.txt to sentence transformers nli example (#1767)

Signed-off-by: Daniel Huang <[email protected]>
huggingface · Feb 12, 2025 · 97beef8 · 97beef8
1 parent 5af3367
commit 97beef8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/examples/sentence-transformers-training/nli/README.md b/examples/sentence-transformers-training/nli/README.md
@@ -4,6 +4,13 @@ Given two sentences (premise and hypothesis), the task of Natural Language Infer
 
 The paper in [Conneau et al.](https://arxiv.org/abs/1705.02364) shows that NLI data can be quite useful when training Sentence Embedding methods. In [Sentence-BERT-Paper](https://arxiv.org/abs/1908.10084) NLI as a first fine-tuning step for sentence embedding methods has been used.
 
+## Requirements
+
+First, you should install the requirements:
+```bash
+pip install -r requirements.txt
+```
+
 # General Models
 
 ## Single-card Training

diff --git a/examples/sentence-transformers-training/nli/requirements.txt b/examples/sentence-transformers-training/nli/requirements.txt
@@ -0,0 +1,2 @@
+datasets
+peft