--extra-index-url https://download.pytorch.org/whl/nightly/cu121
torch==2.1.0.dev20230904+cu121
torchvision==0.16.0.dev20230906
huggingface_hub
OmegaConf
axial-positional-embedding
einops
rotary_embedding_torch
fair-esm
tqdm
importlib
pytorch-lightning==1.9.0