According to https://github.com/huggingface/transformers/issues/34466#issuecomment-2442180500, you need to downgrade PyTorch 2.4 or it will take hours with 2.5
!pip install torch=='2.4.1+cu121' torchvision=='0.19.1+cu121' torchaudio=='2.4.1+cu121' --index-url https://download.pytorch.org/whl/cu121