From 525e6964e843dab51a370824626033155adc2da4 Mon Sep 17 00:00:00 2001 From: Sotirios Anagnostidis Date: Tue, 3 Jan 2023 22:06:59 +0100 Subject: [PATCH] requirements --- model/supervised_finetuning/requirements.txt | 6 ++++++ 1 file changed, 6 insertions(+) create mode 100644 model/supervised_finetuning/requirements.txt diff --git a/model/supervised_finetuning/requirements.txt b/model/supervised_finetuning/requirements.txt new file mode 100644 index 00000000..d579468f --- /dev/null +++ b/model/supervised_finetuning/requirements.txt @@ -0,0 +1,6 @@ +datasets==2.8.0 +numpy==1.23.0 +PyYAML==6.0 +scikit_learn==1.2.0 +torch==1.13.1 +transformers==4.25.1