Using transformers[torch] to limit dependencies
gokceneraslan opened this issue · 1 comments
gokceneraslan commented
I think right now the transformers dependency brings in not only torch but also Jax, tensorflow etc which are not relevant for enformer-pytorch.
I just found out that using transformers[torch] brings only the torch dependency (eg huggingface/transformers#23666). Would it make sense to simply use it in setup.py instead?
lucidrains commented
ah yea sure, let me make that change! i'm guessing your group must be using this repository a lot to request this?