lingyuzeng/cdc_dockerfile
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add evaluate datasets

Browse Source
This commit is contained in:
lingyu zeng
2024-08-28 15:10:58 +08:00
parent a5d1cd0539
commit 299ddc2671
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
finetune/Dockerfile.ngc
View File

@@ -140,6 +140,7 @@ python -m pip install peft tiktoken seaborn blobfile open_clip_torch zstandard m
# optimum 手动解决依赖 # optimum 手动解决依赖
python -m pip install black~=23.1 ruff==0.1.5 diffusers>=0.17.0 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 --no-deps git+https://github.com/huggingface/optimum.git#egg=optimum[diffusers,quality]
python -m pip install evaluate datasets
EOT EOT
RUN <<EOT RUN <<EOT