├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── chatglm_maths ├── __init__.py ├── c00_toy_cpu_train_6b.py ├── c00_toy_gpu_train_6b.py ├── c00_toy_lora_train_6b.py ├── c01_toy_cpu_train_small.py ├── c01_toy_gpu_train_small.py ├── chatglm_6b │ ├── __init__.py │ ├── config.json │ └── tokenizer_config.json ├── math23k_trainset.sample.json ├── models │ ├── README.md │ ├── __init__.py │ ├── configuration_chatglm.py │ ├── modeling_chatglm.py │ ├── ppo_trainer.py │ ├── quantization.py │ └── tokenization_chatglm.py ├── p00_toy_cpu_predit_6b.py ├── p00_toy_lora_predict_6b.py ├── p01_toy_cpu_predict_small.py ├── p10_lora_trl_predict_ppo.py ├── p10_toy_trl_predict_ppo.py ├── t00_tet_chat_chatglm.py ├── t10_lora_trl_train_ppo.py └── t10_toy_trl_train_ppo.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/__init__.py -------------------------------------------------------------------------------- /chatglm_maths/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/__init__.py -------------------------------------------------------------------------------- /chatglm_maths/c00_toy_cpu_train_6b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/c00_toy_cpu_train_6b.py -------------------------------------------------------------------------------- /chatglm_maths/c00_toy_gpu_train_6b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/c00_toy_gpu_train_6b.py -------------------------------------------------------------------------------- /chatglm_maths/c00_toy_lora_train_6b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/c00_toy_lora_train_6b.py -------------------------------------------------------------------------------- /chatglm_maths/c01_toy_cpu_train_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/c01_toy_cpu_train_small.py -------------------------------------------------------------------------------- /chatglm_maths/c01_toy_gpu_train_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/c01_toy_gpu_train_small.py -------------------------------------------------------------------------------- /chatglm_maths/chatglm_6b/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/chatglm_6b/__init__.py -------------------------------------------------------------------------------- /chatglm_maths/chatglm_6b/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/chatglm_6b/config.json -------------------------------------------------------------------------------- /chatglm_maths/chatglm_6b/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/chatglm_6b/tokenizer_config.json -------------------------------------------------------------------------------- /chatglm_maths/math23k_trainset.sample.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/math23k_trainset.sample.json -------------------------------------------------------------------------------- /chatglm_maths/models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/models/README.md -------------------------------------------------------------------------------- /chatglm_maths/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/models/__init__.py -------------------------------------------------------------------------------- /chatglm_maths/models/configuration_chatglm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/models/configuration_chatglm.py -------------------------------------------------------------------------------- /chatglm_maths/models/modeling_chatglm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/models/modeling_chatglm.py -------------------------------------------------------------------------------- /chatglm_maths/models/ppo_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/models/ppo_trainer.py -------------------------------------------------------------------------------- /chatglm_maths/models/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/models/quantization.py -------------------------------------------------------------------------------- /chatglm_maths/models/tokenization_chatglm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/models/tokenization_chatglm.py -------------------------------------------------------------------------------- /chatglm_maths/p00_toy_cpu_predit_6b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/p00_toy_cpu_predit_6b.py -------------------------------------------------------------------------------- /chatglm_maths/p00_toy_lora_predict_6b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/p00_toy_lora_predict_6b.py -------------------------------------------------------------------------------- /chatglm_maths/p01_toy_cpu_predict_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/p01_toy_cpu_predict_small.py -------------------------------------------------------------------------------- /chatglm_maths/p10_lora_trl_predict_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/p10_lora_trl_predict_ppo.py -------------------------------------------------------------------------------- /chatglm_maths/p10_toy_trl_predict_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/p10_toy_trl_predict_ppo.py -------------------------------------------------------------------------------- /chatglm_maths/t00_tet_chat_chatglm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/t00_tet_chat_chatglm.py -------------------------------------------------------------------------------- /chatglm_maths/t10_lora_trl_train_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/t10_lora_trl_train_ppo.py -------------------------------------------------------------------------------- /chatglm_maths/t10_toy_trl_train_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/chatglm_maths/t10_toy_trl_train_ppo.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yongzhuo/chatglm-maths/HEAD/requirements.txt --------------------------------------------------------------------------------