lucidrains/enformer-pytorch

Using transformers[torch] to limit dependencies

gokceneraslan opened this issue · 1 comments

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?

ah yea sure, let me make that change! i'm guessing your group must be using this repository a lot to request this?