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