Removing dependency on `torch` and `torchvision`

paganpasta / eqxvision

A Python package of computer vision models for the Equinox ecosystem.

https://eqxvision.readthedocs.io

MIT License

100 stars 12 forks source link

Removing dependency on `torch` and `torchvision` #15

Open paganpasta opened 2 years ago

paganpasta commented 2 years ago

Currently these are only used to load weights. Is it possible to move the model loading bit into the package so that we dont have to install torch and torchvision. ?