From e0d7808b4e421546c9a4f5b364f72be873d581e8 Mon Sep 17 00:00:00 2001
From: bittersweet1999 <148421775+bittersweet1999@users.noreply.github.com>
Date: Thu, 6 Jun 2024 11:48:07 +0800
Subject: [PATCH] [Fix] fix pip version (#1228)

* fix pip version

* fix pip version
---
 configs/eval_subjective_alpacaeval_official.py | 3 +++
 requirements/extra.txt                         | 3 ++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/configs/eval_subjective_alpacaeval_official.py b/configs/eval_subjective_alpacaeval_official.py
index eac878cc7..f1504fda1 100644
--- a/configs/eval_subjective_alpacaeval_official.py
+++ b/configs/eval_subjective_alpacaeval_official.py
@@ -22,6 +22,9 @@
     reserved_roles=[dict(role='SYSTEM', api_role='SYSTEM')],
 )
 
+
+# To run this config, please ensure to successfully installed `alpaca-eval==0.6` and `scikit-learn==1.5`
+
 # -------------Inference Stage ----------------------------------------
 
 # For subjective evaluation, we often set do sample for models
diff --git a/requirements/extra.txt b/requirements/extra.txt
index 6b3409e74..84f88e6ad 100644
--- a/requirements/extra.txt
+++ b/requirements/extra.txt
@@ -1,2 +1,3 @@
-alpaca-eval
+alpaca-eval==0.6
 faiss_gpu==1.7.2
+scikit-learn==1.5