How can I use TPUs on autotrain

I have been looking to finetune Gemma 2 27b using autotrain and I want to use TPUs instead of GPUs fro the training but I keep getting this error: “CUDA is required but not available for bitsandbytes. Please consider installing the multi-platform enabled version of bitsandbytes, which is currently a work in progress. Please check currently supported platforms and installation instructions at Installation” is there any way I can disable bitsnadbytes for the training?

1 Like

autotrain doesnt support tpu yet. you can open an issue for feature request in github repository :slight_smile:

1 Like