├── README.md ├── data ├── gsm8k.jsonl ├── judge_prompts.jsonl └── mt_bench │ ├── model_judgment │ └── gpt-4_single.jsonl │ ├── question.jsonl │ └── reference_answer │ └── gpt-4.jsonl ├── demo_opt_classic.py ├── demo_opt_eagle.py ├── evaluation ├── eval_opt_classic.py └── eval_opt_eagle.py ├── opt_classic ├── __init__.py ├── configs.py ├── kv_cache.py ├── model.py ├── modeling_llama_kv.py ├── tree.py └── utils.py ├── opt_eagle ├── __init__.py ├── cnets.py ├── configs.py ├── ea_model.py ├── kv_cache.py ├── modeling_llama_kv.py ├── tree.py └── utils.py └── requirements.txt /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/README.md -------------------------------------------------------------------------------- /data/gsm8k.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/data/gsm8k.jsonl -------------------------------------------------------------------------------- /data/judge_prompts.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/data/judge_prompts.jsonl -------------------------------------------------------------------------------- /data/mt_bench/model_judgment/gpt-4_single.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/data/mt_bench/model_judgment/gpt-4_single.jsonl -------------------------------------------------------------------------------- /data/mt_bench/question.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/data/mt_bench/question.jsonl -------------------------------------------------------------------------------- /data/mt_bench/reference_answer/gpt-4.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/data/mt_bench/reference_answer/gpt-4.jsonl -------------------------------------------------------------------------------- /demo_opt_classic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/demo_opt_classic.py -------------------------------------------------------------------------------- /demo_opt_eagle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/demo_opt_eagle.py -------------------------------------------------------------------------------- /evaluation/eval_opt_classic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/evaluation/eval_opt_classic.py -------------------------------------------------------------------------------- /evaluation/eval_opt_eagle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/evaluation/eval_opt_eagle.py -------------------------------------------------------------------------------- /opt_classic/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /opt_classic/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/opt_classic/configs.py -------------------------------------------------------------------------------- /opt_classic/kv_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/opt_classic/kv_cache.py -------------------------------------------------------------------------------- /opt_classic/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/opt_classic/model.py -------------------------------------------------------------------------------- /opt_classic/modeling_llama_kv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/opt_classic/modeling_llama_kv.py -------------------------------------------------------------------------------- /opt_classic/tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/opt_classic/tree.py -------------------------------------------------------------------------------- /opt_classic/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/opt_classic/utils.py -------------------------------------------------------------------------------- /opt_eagle/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /opt_eagle/cnets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/opt_eagle/cnets.py -------------------------------------------------------------------------------- /opt_eagle/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/opt_eagle/configs.py -------------------------------------------------------------------------------- /opt_eagle/ea_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/opt_eagle/ea_model.py -------------------------------------------------------------------------------- /opt_eagle/kv_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/opt_eagle/kv_cache.py -------------------------------------------------------------------------------- /opt_eagle/modeling_llama_kv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/opt_eagle/modeling_llama_kv.py -------------------------------------------------------------------------------- /opt_eagle/tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/opt_eagle/tree.py -------------------------------------------------------------------------------- /opt_eagle/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/opt_eagle/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jikai0Wang/OPT-Tree/HEAD/requirements.txt --------------------------------------------------------------------------------