TabPFN icon indicating copy to clipboard operation
TabPFN copied to clipboard

Support for 'mps' device (i.e. for Apple Silicon)

Open brndnsy opened this issue 1 year ago • 2 comments

It would be great if the model would support Apple Silicon devices so that the 'mps' device could be used as follows: model = TabPFNClassifier(device='mps', N_ensemble_configurations=16)

brndnsy avatar Aug 15 '23 08:08 brndnsy

Sorry, that we never replied for this! We'll keep it in mind, but unfortunately have a lot to do. If you'd like to create a PR for this, would be welcome :)

noahho avatar Nov 08 '23 11:11 noahho

I get an error that MPS device does not support 64-bit floats

ViktorooReps avatar Jul 14 '24 20:07 ViktorooReps