torch==2.2.1 --extra-index-url https://download.pytorch.org/whl/cu121
torchvision==0.17.1 --extra-index-url https://download.pytorch.org/whl/cu121
pytorch-lightning==2.2.1
torch_geometric==2.5.0
torchmetrics==1.3.1
dynaconf==3.2.4
transformers==4.38.2