diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..9b94ab3 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,16 @@ +transformers 4.34.1 +tokenizers 0.14.1 +torchmetrics 1.2.0 +torch 2.1.0 +scikit-learn 1.3.2 +GPUtil 1.4.0 +numpy 1.26.1 +evaluate 0.4.1 +numba 0.58.1 +nltk 3.8.1 +tqdm 4.66.1 +typing 4.8.0 +psycopg2 2.9.9 + + +