fix(benchmarks) Update dependencies for LLM evaluation pipelines (#4287)

adap · Oct 5, 2024 · f696628 · f696628
1 parent 52686d9
commit f696628
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/benchmarks/flowertune-llm/evaluation/code/requirements.txt b/benchmarks/flowertune-llm/evaluation/code/requirements.txt
@@ -4,4 +4,5 @@ evaluate==0.3.0
 sentencepiece==0.2.0
 protobuf==5.27.1
 bitsandbytes==0.43.1
+hf_transfer==0.1.8
 git+https://github.com/bigcode-project/bigcode-evaluation-harness.git@0f3e95f0806e78a4f432056cdb1be93604a51d69
diff --git a/benchmarks/flowertune-llm/evaluation/finance/requirements.txt b/benchmarks/flowertune-llm/evaluation/finance/requirements.txt
@@ -4,3 +4,4 @@ datasets==2.20.0
 sentencepiece==0.2.0
 protobuf==5.27.1
 bitsandbytes==0.43.1
+hf_transfer==0.1.8
diff --git a/benchmarks/flowertune-llm/evaluation/medical/requirements.txt b/benchmarks/flowertune-llm/evaluation/medical/requirements.txt
@@ -5,3 +5,4 @@ datasets==2.20.0
 sentencepiece==0.2.0
 protobuf==5.27.1
 bitsandbytes==0.43.1
+hf_transfer==0.1.8