├── .gitignore ├── README.md ├── README_EN.md ├── datasets └── PromptCBLUE │ └── toy_examples │ ├── dev.json │ ├── dev_structured.json │ ├── results.json │ ├── test.json │ ├── test_predictions.json │ └── train.json ├── peft ├── __init__.py ├── import_utils.py ├── mapping.py ├── peft_model.py ├── tuners │ ├── __init__.py │ ├── adalora.py │ ├── adaption_prompt.py │ ├── lora.py │ ├── p_tuning.py │ ├── prefix_tuning.py │ └── prompt_tuning.py └── utils │ ├── __init__.py │ ├── config.py │ ├── other.py │ └── save_and_load.py ├── pics ├── dingding_groups.jpg ├── promptCBLUE_banner_v0.png ├── promptCBLUE_en_banner_v0.png └── wechat_qrcode.jpg ├── requirements.txt └── src ├── README.md ├── data ├── CBLUE任务改造说明与举例.md ├── templates.json ├── templates_augment.json └── 结构化预测结果格式说明.md ├── download_checkpoints.py ├── evaluation ├── README.txt ├── evaluate.py ├── evaluators.py ├── input_param.json ├── post_generate_process.py └── text2dt_eval_func.py ├── ft_chatglm_lora ├── arguments.py ├── evaluate.sh ├── main.py ├── train.sh ├── trainer.py └── trainer_seq2seq.py ├── ft_chatglm_ptuning ├── arguments.py ├── config.json ├── configuration_chatglm.py ├── evaluate.sh ├── main.py ├── modeling_chatglm.py ├── quantization.py ├── test_modeling_chatglm.py ├── tokenization_chatglm.py ├── tokenizer_config.json ├── train.sh ├── trainer.py └── trainer_seq2seq.py └── ft_llama_lora ├── merge_llama_with_chinese_lora.py ├── run_clm_pt_with_peft.py ├── run_train.sh └── vllm_serving ├── launch_vllm.py ├── llm_engine.py ├── merge_llama_with_lora.py ├── utils.py └── web_service_test.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/README.md -------------------------------------------------------------------------------- /README_EN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/README_EN.md -------------------------------------------------------------------------------- /datasets/PromptCBLUE/toy_examples/dev.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/datasets/PromptCBLUE/toy_examples/dev.json -------------------------------------------------------------------------------- /datasets/PromptCBLUE/toy_examples/dev_structured.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/datasets/PromptCBLUE/toy_examples/dev_structured.json -------------------------------------------------------------------------------- /datasets/PromptCBLUE/toy_examples/results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/datasets/PromptCBLUE/toy_examples/results.json -------------------------------------------------------------------------------- /datasets/PromptCBLUE/toy_examples/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/datasets/PromptCBLUE/toy_examples/test.json -------------------------------------------------------------------------------- /datasets/PromptCBLUE/toy_examples/test_predictions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/datasets/PromptCBLUE/toy_examples/test_predictions.json -------------------------------------------------------------------------------- /datasets/PromptCBLUE/toy_examples/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/datasets/PromptCBLUE/toy_examples/train.json -------------------------------------------------------------------------------- /peft/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/__init__.py -------------------------------------------------------------------------------- /peft/import_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/import_utils.py -------------------------------------------------------------------------------- /peft/mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/mapping.py -------------------------------------------------------------------------------- /peft/peft_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/peft_model.py -------------------------------------------------------------------------------- /peft/tuners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/tuners/__init__.py -------------------------------------------------------------------------------- /peft/tuners/adalora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/tuners/adalora.py -------------------------------------------------------------------------------- /peft/tuners/adaption_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/tuners/adaption_prompt.py -------------------------------------------------------------------------------- /peft/tuners/lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/tuners/lora.py -------------------------------------------------------------------------------- /peft/tuners/p_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/tuners/p_tuning.py -------------------------------------------------------------------------------- /peft/tuners/prefix_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/tuners/prefix_tuning.py -------------------------------------------------------------------------------- /peft/tuners/prompt_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/tuners/prompt_tuning.py -------------------------------------------------------------------------------- /peft/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/utils/__init__.py -------------------------------------------------------------------------------- /peft/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/utils/config.py -------------------------------------------------------------------------------- /peft/utils/other.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/utils/other.py -------------------------------------------------------------------------------- /peft/utils/save_and_load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/peft/utils/save_and_load.py -------------------------------------------------------------------------------- /pics/dingding_groups.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/pics/dingding_groups.jpg -------------------------------------------------------------------------------- /pics/promptCBLUE_banner_v0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/pics/promptCBLUE_banner_v0.png -------------------------------------------------------------------------------- /pics/promptCBLUE_en_banner_v0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/pics/promptCBLUE_en_banner_v0.png -------------------------------------------------------------------------------- /pics/wechat_qrcode.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/pics/wechat_qrcode.jpg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/README.md -------------------------------------------------------------------------------- /src/data/CBLUE任务改造说明与举例.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/data/CBLUE任务改造说明与举例.md -------------------------------------------------------------------------------- /src/data/templates.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/data/templates.json -------------------------------------------------------------------------------- /src/data/templates_augment.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/data/templates_augment.json -------------------------------------------------------------------------------- /src/data/结构化预测结果格式说明.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/data/结构化预测结果格式说明.md -------------------------------------------------------------------------------- /src/download_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/download_checkpoints.py -------------------------------------------------------------------------------- /src/evaluation/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/evaluation/README.txt -------------------------------------------------------------------------------- /src/evaluation/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/evaluation/evaluate.py -------------------------------------------------------------------------------- /src/evaluation/evaluators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/evaluation/evaluators.py -------------------------------------------------------------------------------- /src/evaluation/input_param.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/evaluation/input_param.json -------------------------------------------------------------------------------- /src/evaluation/post_generate_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/evaluation/post_generate_process.py -------------------------------------------------------------------------------- /src/evaluation/text2dt_eval_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/evaluation/text2dt_eval_func.py -------------------------------------------------------------------------------- /src/ft_chatglm_lora/arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_lora/arguments.py -------------------------------------------------------------------------------- /src/ft_chatglm_lora/evaluate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_lora/evaluate.sh -------------------------------------------------------------------------------- /src/ft_chatglm_lora/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_lora/main.py -------------------------------------------------------------------------------- /src/ft_chatglm_lora/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_lora/train.sh -------------------------------------------------------------------------------- /src/ft_chatglm_lora/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_lora/trainer.py -------------------------------------------------------------------------------- /src/ft_chatglm_lora/trainer_seq2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_lora/trainer_seq2seq.py -------------------------------------------------------------------------------- /src/ft_chatglm_ptuning/arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_ptuning/arguments.py -------------------------------------------------------------------------------- /src/ft_chatglm_ptuning/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_ptuning/config.json -------------------------------------------------------------------------------- /src/ft_chatglm_ptuning/configuration_chatglm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_ptuning/configuration_chatglm.py -------------------------------------------------------------------------------- /src/ft_chatglm_ptuning/evaluate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_ptuning/evaluate.sh -------------------------------------------------------------------------------- /src/ft_chatglm_ptuning/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_ptuning/main.py -------------------------------------------------------------------------------- /src/ft_chatglm_ptuning/modeling_chatglm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_ptuning/modeling_chatglm.py -------------------------------------------------------------------------------- /src/ft_chatglm_ptuning/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_ptuning/quantization.py -------------------------------------------------------------------------------- /src/ft_chatglm_ptuning/test_modeling_chatglm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_ptuning/test_modeling_chatglm.py -------------------------------------------------------------------------------- /src/ft_chatglm_ptuning/tokenization_chatglm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_ptuning/tokenization_chatglm.py -------------------------------------------------------------------------------- /src/ft_chatglm_ptuning/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_ptuning/tokenizer_config.json -------------------------------------------------------------------------------- /src/ft_chatglm_ptuning/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_ptuning/train.sh -------------------------------------------------------------------------------- /src/ft_chatglm_ptuning/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_ptuning/trainer.py -------------------------------------------------------------------------------- /src/ft_chatglm_ptuning/trainer_seq2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_chatglm_ptuning/trainer_seq2seq.py -------------------------------------------------------------------------------- /src/ft_llama_lora/merge_llama_with_chinese_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_llama_lora/merge_llama_with_chinese_lora.py -------------------------------------------------------------------------------- /src/ft_llama_lora/run_clm_pt_with_peft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_llama_lora/run_clm_pt_with_peft.py -------------------------------------------------------------------------------- /src/ft_llama_lora/run_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_llama_lora/run_train.sh -------------------------------------------------------------------------------- /src/ft_llama_lora/vllm_serving/launch_vllm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_llama_lora/vllm_serving/launch_vllm.py -------------------------------------------------------------------------------- /src/ft_llama_lora/vllm_serving/llm_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_llama_lora/vllm_serving/llm_engine.py -------------------------------------------------------------------------------- /src/ft_llama_lora/vllm_serving/merge_llama_with_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_llama_lora/vllm_serving/merge_llama_with_lora.py -------------------------------------------------------------------------------- /src/ft_llama_lora/vllm_serving/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_llama_lora/vllm_serving/utils.py -------------------------------------------------------------------------------- /src/ft_llama_lora/vllm_serving/web_service_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-wzhu/PromptCBLUE/HEAD/src/ft_llama_lora/vllm_serving/web_service_test.py --------------------------------------------------------------------------------