From 299ddc267145955e2520eb7b807c96e6f3ea8f0d Mon Sep 17 00:00:00 2001
From: lingyu zeng <ze.ga@qq.com>
Date: Wed, 28 Aug 2024 15:10:58 +0800
Subject: [PATCH] add evaluate datasets

---
 finetune/Dockerfile.ngc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/finetune/Dockerfile.ngc b/finetune/Dockerfile.ngc
index 4fb183b..bfd89cb 100644
--- a/finetune/Dockerfile.ngc
+++ b/finetune/Dockerfile.ngc
@@ -140,6 +140,7 @@ python -m pip install peft tiktoken seaborn blobfile open_clip_torch zstandard m
 # optimum 手动解决依赖
 python -m pip install black~=23.1 ruff==0.1.5 diffusers>=0.17.0
 python -m pip install --no-deps git+https://github.com/huggingface/optimum.git#egg=optimum[diffusers,quality]
+python -m pip install evaluate datasets
 EOT
 
 RUN <<EOT