Add requirements.txt
Browse files- requirements.txt +8 -0
requirements.txt
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Verified against the versions present on the development machine
|
| 2 |
+
# (Python 3.11.9, CUDA 12.4, RTX 4060 8 GB).
|
| 3 |
+
torch>=2.4
|
| 4 |
+
transformers>=4.48 # ModernBERT support landed in 4.48
|
| 5 |
+
scikit-learn>=1.3
|
| 6 |
+
numpy>=1.24,<2.0
|
| 7 |
+
pytest>=7.4
|
| 8 |
+
huggingface_hub>=0.24
|