Open wangkuiyi opened 1 year ago
@silvasean do you know the original reason for limiting the pillow version?
It is documented in the comment just above the restriction.
# For torchvision, use pillow<7 to avoid `ImportError: cannot import name 'PILLOW_VERSION' from 'PIL'`
# See https://github.com/pytorch/vision/issues/1712
If that doesn't happen anymore then we can remove the limitation (we should remove that commend too though)
Fix https://github.com/iree-org/iree-torch/issues/78
After this change, I successfully install dependencies with
python -m pip install -r requirements.txt
And I can build iree-torch with
python setup.py develop
, so thatpython -c 'import iree_torch'
works for me.However, when I ran
python examples/bert.py
, the following linehttps://github.com/iree-org/iree-torch/blob/5fbaf505384eef01f63a618bcc4e59929b87f676/examples/bert.py#L21
raised the error.
pip install transformers
saved me. But I guess this fix might go into another pull request? I made https://github.com/iree-org/iree-torch/pull/80