Skip to content

Commit

Permalink
Add missing dataset dependency
Browse files Browse the repository at this point in the history
  • Loading branch information
mudler committed Mar 6, 2024
1 parent 293be13 commit 72d1cd2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion backend/python/common-env/transformers/install.sh
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ if [ -d "/opt/intel" ]; then
# Intel GPU: If the directory exists, we assume we are using the intel image
# (no conda env)
# https://github.com/intel/intel-extension-for-pytorch/issues/538
pip install intel-extension-for-transformers
pip install intel-extension-for-transformers datasets
fi

if [ "$PIP_CACHE_PURGE" = true ] ; then
Expand Down

0 comments on commit 72d1cd2

Please sign in to comment.