pip install torch==2.1.0

pip install pytorch-fast-transformers==0.4.0

pip install -r requirements.txt