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