├── .gitignore ├── LICENSE ├── README.md ├── assets └── .gitkeep ├── images └── sketch8.gif ├── requirements.txt ├── scripts ├── eval_td.py ├── eval_td_search.py ├── test_setup.py ├── train.py ├── train_ar.py └── train_value.py └── td ├── data ├── data_loader.py └── treequeues.py ├── environments ├── __init__.py ├── blockworld.py ├── csg2d.py ├── csg2da.py ├── csg2dh.py ├── environment.py ├── goal_checker.py ├── nanosvg.py ├── rainbow.py ├── tinysvg.py └── tinysvgoffset.py ├── grammar.py ├── learning ├── constrained_decoding.py ├── evaluation.py ├── gpt.py ├── tokenizer.py └── utils.py ├── notebooks ├── .gitignore └── environment_diagnosis.ipynb └── samplers ├── __init__.py └── mutator.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/README.md -------------------------------------------------------------------------------- /assets/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /images/sketch8.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/images/sketch8.gif -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/eval_td.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/scripts/eval_td.py -------------------------------------------------------------------------------- /scripts/eval_td_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/scripts/eval_td_search.py -------------------------------------------------------------------------------- /scripts/test_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/scripts/test_setup.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/scripts/train.py -------------------------------------------------------------------------------- /scripts/train_ar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/scripts/train_ar.py -------------------------------------------------------------------------------- /scripts/train_value.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/scripts/train_value.py -------------------------------------------------------------------------------- /td/data/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/data/data_loader.py -------------------------------------------------------------------------------- /td/data/treequeues.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/data/treequeues.py -------------------------------------------------------------------------------- /td/environments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/environments/__init__.py -------------------------------------------------------------------------------- /td/environments/blockworld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/environments/blockworld.py -------------------------------------------------------------------------------- /td/environments/csg2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/environments/csg2d.py -------------------------------------------------------------------------------- /td/environments/csg2da.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/environments/csg2da.py -------------------------------------------------------------------------------- /td/environments/csg2dh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/environments/csg2dh.py -------------------------------------------------------------------------------- /td/environments/environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/environments/environment.py -------------------------------------------------------------------------------- /td/environments/goal_checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/environments/goal_checker.py -------------------------------------------------------------------------------- /td/environments/nanosvg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/environments/nanosvg.py -------------------------------------------------------------------------------- /td/environments/rainbow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/environments/rainbow.py -------------------------------------------------------------------------------- /td/environments/tinysvg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/environments/tinysvg.py -------------------------------------------------------------------------------- /td/environments/tinysvgoffset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/environments/tinysvgoffset.py -------------------------------------------------------------------------------- /td/grammar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/grammar.py -------------------------------------------------------------------------------- /td/learning/constrained_decoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/learning/constrained_decoding.py -------------------------------------------------------------------------------- /td/learning/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/learning/evaluation.py -------------------------------------------------------------------------------- /td/learning/gpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/learning/gpt.py -------------------------------------------------------------------------------- /td/learning/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/learning/tokenizer.py -------------------------------------------------------------------------------- /td/learning/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/learning/utils.py -------------------------------------------------------------------------------- /td/notebooks/.gitignore: -------------------------------------------------------------------------------- 1 | playground/ 2 | -------------------------------------------------------------------------------- /td/notebooks/environment_diagnosis.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/notebooks/environment_diagnosis.ipynb -------------------------------------------------------------------------------- /td/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/samplers/__init__.py -------------------------------------------------------------------------------- /td/samplers/mutator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/revalo/tree-diffusion/HEAD/td/samplers/mutator.py --------------------------------------------------------------------------------