Provide the option for installing Ensemble-TabPFN with or without GPU support

ersilia-os / chempfn

Ensemble-based, size-agnostic wrapper for the TabPFN classifier

GNU General Public License v3.0

28 stars 0 forks source link

Provide the option for installing Ensemble-TabPFN with or without GPU support #25

Open DhanshreeA opened 1 year ago

DhanshreeA commented 1 year ago

Currently the original TabPFN installation assumes that the user may have a GPU to spare and installs a bunch of cuda libraries. This makes the overall installation very bulky, and pointless if the user does not have a GPU. A solution is to create a tabpfn fork and install this dependency directly from source, supporting both options.