From 601c4d395cbea95ebc751e5a76bbbf27cdaacd1b Mon Sep 17 00:00:00 2001 From: Yi Wang Date: Sun, 22 Jan 2023 16:36:13 -0800 Subject: [PATCH] Add transformers into requirements.txt examples/bert.py needs the package. --- requirements.txt | 3 +++ 1 file changed, 3 insertions(+) diff --git a/requirements.txt b/requirements.txt index 709aeb1..b0cf41a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -17,6 +17,9 @@ torch torchvision numpy +# For examples/bert.py +transformers + # Testing # For torchvision, use pillow<7 to avoid `ImportError: cannot import name 'PILLOW_VERSION' from 'PIL'` # See https://github.com/pytorch/vision/issues/1712