├── README.md ├── data └── data_128.jsonl ├── main.py ├── patch ├── EABF.py ├── __init__.py └── flash_attn_monkey_patch.py ├── prompt.py └── retrieve_attn.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAIR-NLP/Entropy-ABF/HEAD/README.md -------------------------------------------------------------------------------- /data/data_128.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAIR-NLP/Entropy-ABF/HEAD/data/data_128.jsonl -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAIR-NLP/Entropy-ABF/HEAD/main.py -------------------------------------------------------------------------------- /patch/EABF.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAIR-NLP/Entropy-ABF/HEAD/patch/EABF.py -------------------------------------------------------------------------------- /patch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAIR-NLP/Entropy-ABF/HEAD/patch/__init__.py -------------------------------------------------------------------------------- /patch/flash_attn_monkey_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAIR-NLP/Entropy-ABF/HEAD/patch/flash_attn_monkey_patch.py -------------------------------------------------------------------------------- /prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAIR-NLP/Entropy-ABF/HEAD/prompt.py -------------------------------------------------------------------------------- /retrieve_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAIR-NLP/Entropy-ABF/HEAD/retrieve_attn.py --------------------------------------------------------------------------------