text-graph-generalization

Code for our paper Compositional Generalization with Grounded Language Models
git clone https://esimon.eu/repos/text-graph-generalization.git
Log | Files | Refs | README | LICENSE

requirements.txt (241B)

      1 jsonlines==4.0.0
      2 networkx==3.2.1
      3 numpy==1.26.3
      4 safetensors==0.4.2
      5 tokenizers==0.15.1
      6 torch==1.11.0+rocm4.5.2
      7 torch-cluster==1.6.3
      8 torch-scatter==2.1.2
      9 torch-sparse==0.6.18
     10 torch-spline-conv==1.2.2
     11 torch_geometric==2.4.0
     12 transformers==4.37.1