Skip to content

Commit 676b47f

Browse files
Harden required phrase model training
Signed-off-by: Kaushik Kumar <kaushikrjpm10@gmail.com>
1 parent 4ccfe39 commit 676b47f

7 files changed

Lines changed: 4073 additions & 697 deletions

File tree

etc/requirements-ml.txt

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,12 @@
22
# Install with: pip install -r etc/requirements-ml.txt
33

44
torch>=2.0
5-
transformers>=4.44
5+
transformers==4.57.3
6+
huggingface-hub==0.36.2
67
sentencepiece>=0.2
78
protobuf>=3.20
89
accelerate>=0.33
9-
pytorch-crf>=0.7.2
10+
pytorch-crf==0.7.2
1011
safetensors>=0.4
1112

1213
# Used explicitly for DeBERTa-large training on a 16 GB GPU.

0 commit comments

Comments
 (0)