text-graph-generalization, branch HEAD Code for our paper Compositional Generalization with Grounded Language Models