torch==1.10.0 transformers==4.25.1 sentencepiece pytorch-lightning==1.6.4