From 6d92ce6fdd8c3c91a41c226cb789c473c436fcd1 Mon Sep 17 00:00:00 2001 From: Mark Kurtz Date: Fri, 30 Jul 2021 11:07:28 -0400 Subject: [PATCH] Fix transformers not installing correct dependencies --- integrations/huggingface-transformers/setup_integration.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/integrations/huggingface-transformers/setup_integration.sh b/integrations/huggingface-transformers/setup_integration.sh index 7a851c50893..1cf6227c524 100644 --- a/integrations/huggingface-transformers/setup_integration.sh +++ b/integrations/huggingface-transformers/setup_integration.sh @@ -8,3 +8,4 @@ git clone https://github.com/neuralmagic/transformers.git cd transformers pip install -e . pip install datasets +pip install sparseml[torch]