From d3e0e429a26337ed2aab77826a78d3c8b5026c0c Mon Sep 17 00:00:00 2001
From: Tommaso Tarchi <102803908+TommasoTarchi@users.noreply.github.com>
Date: Tue, 30 Apr 2024 14:58:39 +0200
Subject: [PATCH] Added package versions to requirements.txt

After meeting several compatibility issues while setting up the virtual environment for running the model, I retrieved the specific versions of packages and added to requirements.txt. Now inference and finetuning work without any problem.
However, I was not able to install a compatible version of Triton.
---
 requirements.txt | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 8ecb3cd..9f64441 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,8 @@
-einops
+einops==0.6.1
 transformers==4.29.2
-peft
-omegaconf
-torch
-evaluate
-accelerate
+peft==0.3.0
+omegaconf==2.3.0
+torch==1.13.1
+evaluate==0.4.0
+accelerate==0.20.3
+scikit-learn==1.2.2