gitextract_x8c_d3k7/ ├── README.md ├── data/ │ ├── __init__.py │ ├── arxiv/ │ │ ├── artificial intelligence_10047_15000_15_abs.txt │ │ ├── artificial intelligence_10047_15000_15_title.txt │ │ ├── artificial intelligence_134_15000_200_abs.txt │ │ ├── artificial intelligence_134_15000_200_title.txt │ │ ├── computer vision_14582_15000_15_abs.txt │ │ ├── computer vision_14582_15000_15_title.txt │ │ ├── language generation_14514_15000_15_abs.txt │ │ └── language generation_14514_15000_15_title.txt │ ├── arxiv_dataset.py │ ├── plot_dataset.py │ ├── prompt_dataset.py │ ├── util.py │ └── yelp_dataset.py ├── dist_utils.py ├── eval_ppl.py ├── eval_ppl_prefix.py ├── generate.py ├── generate_prefix.py ├── model.py ├── train.py ├── train_dist.py ├── train_dist_half.py ├── tsne_plot.py └── util.py