From 72d1cd257b2a805620ba667888fad5b1f84c383d Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Wed, 6 Mar 2024 17:11:03 +0100 Subject: [PATCH] Add missing dataset dependency --- backend/python/common-env/transformers/install.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/python/common-env/transformers/install.sh b/backend/python/common-env/transformers/install.sh index 4df682abd404..169a223197fd 100644 --- a/backend/python/common-env/transformers/install.sh +++ b/backend/python/common-env/transformers/install.sh @@ -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