add evaluate datasets

2024-08-28 15:10:58 +08:00
parent a5d1cd0539
commit 299ddc2671
1 changed files with 1 additions and 0 deletions
--- 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