requirements.txt

torch==1.12.1+cu113
transformers==4.29.1
einops
tqdm
wandb
rotary-embedding-torch
biopython
scipy
dm-tree
matplotlib
seaborn
black
ipython
torchtyping
--extra-index-url https://download.pytorch.org/whl/cu113