torch @ https://download.pytorch.org/whl/cu121_full/torch-2.5.0%2Bcu121-cp310-cp310-linux_x86_64.whl transformers==4.46.1 TensorFlow==2.18.0 tf-keras==2.18.0