requirements.txt

networkx==2.8.7
numpy==1.23
omegaconf==2.3.0
pytorch_lightning==2.0.4
torch_geometric==2.3.1
torchmetrics==0.11.4
tqdm==4.65.0
torch==2.3.0
torchvision==0.18.0
torchaudio==2.3.0
gradio==4.32.0
wandb==0.15.4
sentence-transformers==2.6.1
PyGSP==0.5.1
pyemd==1.0.0
accelerate==0.30.1