├── .gitignore ├── CODE_OF_CONDUCT.md ├── LICENSE ├── NOTICE.md ├── README.md ├── SECURITY.md ├── SUPPORT.md ├── inference_code ├── evaluation.py ├── qlora_merge.py ├── utils.py └── vllm_inference.py ├── pictures └── overall.png ├── prompts ├── prompt_correction_csqa.txt ├── prompt_correction_gsm8k.txt └── prompt_correction_math.txt ├── requirements.txt ├── test_data ├── csqa_dev.jsonl ├── gsm8k_test.jsonl ├── gsm8k_test_wizardmath_prompt.jsonl ├── math_test.jsonl └── math_test_wizardmath_prompt.jsonl └── training_code ├── llama_2_70b_gsm8k.sh └── qlora.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/.gitignore -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/NOTICE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/README.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/SECURITY.md -------------------------------------------------------------------------------- /SUPPORT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/SUPPORT.md -------------------------------------------------------------------------------- /inference_code/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/inference_code/evaluation.py -------------------------------------------------------------------------------- /inference_code/qlora_merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/inference_code/qlora_merge.py -------------------------------------------------------------------------------- /inference_code/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/inference_code/utils.py -------------------------------------------------------------------------------- /inference_code/vllm_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/inference_code/vllm_inference.py -------------------------------------------------------------------------------- /pictures/overall.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/pictures/overall.png -------------------------------------------------------------------------------- /prompts/prompt_correction_csqa.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/prompts/prompt_correction_csqa.txt -------------------------------------------------------------------------------- /prompts/prompt_correction_gsm8k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/prompts/prompt_correction_gsm8k.txt -------------------------------------------------------------------------------- /prompts/prompt_correction_math.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/prompts/prompt_correction_math.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/requirements.txt -------------------------------------------------------------------------------- /test_data/csqa_dev.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/test_data/csqa_dev.jsonl -------------------------------------------------------------------------------- /test_data/gsm8k_test.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/test_data/gsm8k_test.jsonl -------------------------------------------------------------------------------- /test_data/gsm8k_test_wizardmath_prompt.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/test_data/gsm8k_test_wizardmath_prompt.jsonl -------------------------------------------------------------------------------- /test_data/math_test.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/test_data/math_test.jsonl -------------------------------------------------------------------------------- /test_data/math_test_wizardmath_prompt.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/test_data/math_test_wizardmath_prompt.jsonl -------------------------------------------------------------------------------- /training_code/llama_2_70b_gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/training_code/llama_2_70b_gsm8k.sh -------------------------------------------------------------------------------- /training_code/qlora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/LEMA/HEAD/training_code/qlora.py --------------------------------------------------------------------------------