├── README.md ├── cache └── user.config ├── data └── dataset_info.json ├── evaluation ├── ceval │ ├── ceval.py │ ├── ceval.zip │ └── mapping.json ├── cmmlu │ ├── cmmlu.py │ ├── cmmlu.zip │ └── mapping.json └── mmlu │ ├── mapping.json │ ├── mmlu.py │ └── mmlu.zip ├── pics └── logo.jpg ├── requirements.txt ├── setup.py ├── src ├── api_demo.py ├── cli_demo.py ├── evaluate.py ├── export_model.py ├── llmtuner │ ├── __init__.py │ ├── api │ │ ├── __init__.py │ │ ├── app.py │ │ └── protocol.py │ ├── chat │ │ ├── __init__.py │ │ └── chat_model.py │ ├── data │ │ ├── __init__.py │ │ ├── loader.py │ │ ├── preprocess.py │ │ ├── template.py │ │ └── utils.py │ ├── eval │ │ ├── __init__.py │ │ ├── evaluator.py │ │ └── template.py │ ├── extras │ │ ├── __init__.py │ │ ├── callbacks.py │ │ ├── constants.py │ │ ├── logging.py │ │ ├── misc.py │ │ ├── packages.py │ │ ├── patches │ │ │ ├── __init__.py │ │ │ └── llama_patch.py │ │ └── ploting.py │ ├── hparams │ │ ├── __init__.py │ │ ├── data_args.py │ │ ├── evaluation_args.py │ │ ├── finetuning_args.py │ │ ├── generating_args.py │ │ └── model_args.py │ ├── model │ │ ├── __init__.py │ │ ├── adapter.py │ │ ├── loader.py │ │ ├── parser.py │ │ ├── patcher.py │ │ └── utils.py │ ├── train │ │ ├── __init__.py │ │ ├── dpo │ │ │ ├── __init__.py │ │ │ ├── collator.py │ │ │ ├── trainer.py │ │ │ └── workflow.py │ │ ├── ppo │ │ │ ├── __init__.py │ │ │ ├── trainer.py │ │ │ ├── utils.py │ │ │ └── workflow.py │ │ ├── pt │ │ │ ├── __init__.py │ │ │ └── workflow.py │ │ ├── rm │ │ │ ├── __init__.py │ │ │ ├── collator.py │ │ │ ├── metric.py │ │ │ ├── trainer.py │ │ │ └── workflow.py │ │ ├── sft │ │ │ ├── __init__.py │ │ │ ├── metric.py │ │ │ ├── trainer.py │ │ │ └── workflow.py │ │ ├── tuner.py │ │ └── utils.py │ └── webui │ │ ├── __init__.py │ │ ├── chatter.py │ │ ├── common.py │ │ ├── components │ │ ├── __init__.py │ │ ├── chatbot.py │ │ ├── data.py │ │ ├── eval.py │ │ ├── export.py │ │ ├── infer.py │ │ ├── top.py │ │ └── train.py │ │ ├── css.py │ │ ├── engine.py │ │ ├── interface.py │ │ ├── locales.py │ │ ├── manager.py │ │ ├── runner.py │ │ └── utils.py ├── train_bash.py ├── train_web.py └── web_demo.py ├── start_train.sh ├── start_web_demo.sh └── tests ├── cal_flops.py ├── cal_lr.py ├── llamafy_baichuan2.py ├── llamafy_qwen.py ├── loftq_init.py └── quantize.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/README.md -------------------------------------------------------------------------------- /cache/user.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/cache/user.config -------------------------------------------------------------------------------- /data/dataset_info.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/data/dataset_info.json -------------------------------------------------------------------------------- /evaluation/ceval/ceval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/evaluation/ceval/ceval.py -------------------------------------------------------------------------------- /evaluation/ceval/ceval.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/evaluation/ceval/ceval.zip -------------------------------------------------------------------------------- /evaluation/ceval/mapping.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/evaluation/ceval/mapping.json -------------------------------------------------------------------------------- /evaluation/cmmlu/cmmlu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/evaluation/cmmlu/cmmlu.py -------------------------------------------------------------------------------- /evaluation/cmmlu/cmmlu.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/evaluation/cmmlu/cmmlu.zip -------------------------------------------------------------------------------- /evaluation/cmmlu/mapping.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/evaluation/cmmlu/mapping.json -------------------------------------------------------------------------------- /evaluation/mmlu/mapping.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/evaluation/mmlu/mapping.json -------------------------------------------------------------------------------- /evaluation/mmlu/mmlu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/evaluation/mmlu/mmlu.py -------------------------------------------------------------------------------- /evaluation/mmlu/mmlu.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/evaluation/mmlu/mmlu.zip -------------------------------------------------------------------------------- /pics/logo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/pics/logo.jpg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/setup.py -------------------------------------------------------------------------------- /src/api_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/api_demo.py -------------------------------------------------------------------------------- /src/cli_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/cli_demo.py -------------------------------------------------------------------------------- /src/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/evaluate.py -------------------------------------------------------------------------------- /src/export_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/export_model.py -------------------------------------------------------------------------------- /src/llmtuner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/__init__.py -------------------------------------------------------------------------------- /src/llmtuner/api/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/api/__init__.py -------------------------------------------------------------------------------- /src/llmtuner/api/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/api/app.py -------------------------------------------------------------------------------- /src/llmtuner/api/protocol.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/api/protocol.py -------------------------------------------------------------------------------- /src/llmtuner/chat/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/chat/__init__.py -------------------------------------------------------------------------------- /src/llmtuner/chat/chat_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/chat/chat_model.py -------------------------------------------------------------------------------- /src/llmtuner/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/data/__init__.py -------------------------------------------------------------------------------- /src/llmtuner/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/data/loader.py -------------------------------------------------------------------------------- /src/llmtuner/data/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/data/preprocess.py -------------------------------------------------------------------------------- /src/llmtuner/data/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/data/template.py -------------------------------------------------------------------------------- /src/llmtuner/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/data/utils.py -------------------------------------------------------------------------------- /src/llmtuner/eval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/eval/__init__.py -------------------------------------------------------------------------------- /src/llmtuner/eval/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/eval/evaluator.py -------------------------------------------------------------------------------- /src/llmtuner/eval/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/eval/template.py -------------------------------------------------------------------------------- /src/llmtuner/extras/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/llmtuner/extras/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/extras/callbacks.py -------------------------------------------------------------------------------- /src/llmtuner/extras/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/extras/constants.py -------------------------------------------------------------------------------- /src/llmtuner/extras/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/extras/logging.py -------------------------------------------------------------------------------- /src/llmtuner/extras/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/extras/misc.py -------------------------------------------------------------------------------- /src/llmtuner/extras/packages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/extras/packages.py -------------------------------------------------------------------------------- /src/llmtuner/extras/patches/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/llmtuner/extras/patches/llama_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/extras/patches/llama_patch.py -------------------------------------------------------------------------------- /src/llmtuner/extras/ploting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/extras/ploting.py -------------------------------------------------------------------------------- /src/llmtuner/hparams/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/hparams/__init__.py -------------------------------------------------------------------------------- /src/llmtuner/hparams/data_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/hparams/data_args.py -------------------------------------------------------------------------------- /src/llmtuner/hparams/evaluation_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/hparams/evaluation_args.py -------------------------------------------------------------------------------- /src/llmtuner/hparams/finetuning_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/hparams/finetuning_args.py -------------------------------------------------------------------------------- /src/llmtuner/hparams/generating_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/hparams/generating_args.py -------------------------------------------------------------------------------- /src/llmtuner/hparams/model_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/hparams/model_args.py -------------------------------------------------------------------------------- /src/llmtuner/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/model/__init__.py -------------------------------------------------------------------------------- /src/llmtuner/model/adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/model/adapter.py -------------------------------------------------------------------------------- /src/llmtuner/model/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/model/loader.py -------------------------------------------------------------------------------- /src/llmtuner/model/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/model/parser.py -------------------------------------------------------------------------------- /src/llmtuner/model/patcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/model/patcher.py -------------------------------------------------------------------------------- /src/llmtuner/model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/model/utils.py -------------------------------------------------------------------------------- /src/llmtuner/train/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/__init__.py -------------------------------------------------------------------------------- /src/llmtuner/train/dpo/__init__.py: -------------------------------------------------------------------------------- 1 | from llmtuner.train.dpo.workflow import run_dpo 2 | -------------------------------------------------------------------------------- /src/llmtuner/train/dpo/collator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/dpo/collator.py -------------------------------------------------------------------------------- /src/llmtuner/train/dpo/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/dpo/trainer.py -------------------------------------------------------------------------------- /src/llmtuner/train/dpo/workflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/dpo/workflow.py -------------------------------------------------------------------------------- /src/llmtuner/train/ppo/__init__.py: -------------------------------------------------------------------------------- 1 | from llmtuner.train.ppo.workflow import run_ppo 2 | -------------------------------------------------------------------------------- /src/llmtuner/train/ppo/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/ppo/trainer.py -------------------------------------------------------------------------------- /src/llmtuner/train/ppo/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/ppo/utils.py -------------------------------------------------------------------------------- /src/llmtuner/train/ppo/workflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/ppo/workflow.py -------------------------------------------------------------------------------- /src/llmtuner/train/pt/__init__.py: -------------------------------------------------------------------------------- 1 | from llmtuner.train.pt.workflow import run_pt 2 | -------------------------------------------------------------------------------- /src/llmtuner/train/pt/workflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/pt/workflow.py -------------------------------------------------------------------------------- /src/llmtuner/train/rm/__init__.py: -------------------------------------------------------------------------------- 1 | from llmtuner.train.rm.workflow import run_rm 2 | -------------------------------------------------------------------------------- /src/llmtuner/train/rm/collator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/rm/collator.py -------------------------------------------------------------------------------- /src/llmtuner/train/rm/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/rm/metric.py -------------------------------------------------------------------------------- /src/llmtuner/train/rm/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/rm/trainer.py -------------------------------------------------------------------------------- /src/llmtuner/train/rm/workflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/rm/workflow.py -------------------------------------------------------------------------------- /src/llmtuner/train/sft/__init__.py: -------------------------------------------------------------------------------- 1 | from llmtuner.train.sft.workflow import run_sft 2 | -------------------------------------------------------------------------------- /src/llmtuner/train/sft/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/sft/metric.py -------------------------------------------------------------------------------- /src/llmtuner/train/sft/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/sft/trainer.py -------------------------------------------------------------------------------- /src/llmtuner/train/sft/workflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/sft/workflow.py -------------------------------------------------------------------------------- /src/llmtuner/train/tuner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/tuner.py -------------------------------------------------------------------------------- /src/llmtuner/train/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/train/utils.py -------------------------------------------------------------------------------- /src/llmtuner/webui/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/__init__.py -------------------------------------------------------------------------------- /src/llmtuner/webui/chatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/chatter.py -------------------------------------------------------------------------------- /src/llmtuner/webui/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/common.py -------------------------------------------------------------------------------- /src/llmtuner/webui/components/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/components/__init__.py -------------------------------------------------------------------------------- /src/llmtuner/webui/components/chatbot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/components/chatbot.py -------------------------------------------------------------------------------- /src/llmtuner/webui/components/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/components/data.py -------------------------------------------------------------------------------- /src/llmtuner/webui/components/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/components/eval.py -------------------------------------------------------------------------------- /src/llmtuner/webui/components/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/components/export.py -------------------------------------------------------------------------------- /src/llmtuner/webui/components/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/components/infer.py -------------------------------------------------------------------------------- /src/llmtuner/webui/components/top.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/components/top.py -------------------------------------------------------------------------------- /src/llmtuner/webui/components/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/components/train.py -------------------------------------------------------------------------------- /src/llmtuner/webui/css.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/css.py -------------------------------------------------------------------------------- /src/llmtuner/webui/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/engine.py -------------------------------------------------------------------------------- /src/llmtuner/webui/interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/interface.py -------------------------------------------------------------------------------- /src/llmtuner/webui/locales.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/locales.py -------------------------------------------------------------------------------- /src/llmtuner/webui/manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/manager.py -------------------------------------------------------------------------------- /src/llmtuner/webui/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/runner.py -------------------------------------------------------------------------------- /src/llmtuner/webui/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/llmtuner/webui/utils.py -------------------------------------------------------------------------------- /src/train_bash.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/train_bash.py -------------------------------------------------------------------------------- /src/train_web.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/train_web.py -------------------------------------------------------------------------------- /src/web_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/src/web_demo.py -------------------------------------------------------------------------------- /start_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/start_train.sh -------------------------------------------------------------------------------- /start_web_demo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/start_web_demo.sh -------------------------------------------------------------------------------- /tests/cal_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/tests/cal_flops.py -------------------------------------------------------------------------------- /tests/cal_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/tests/cal_lr.py -------------------------------------------------------------------------------- /tests/llamafy_baichuan2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/tests/llamafy_baichuan2.py -------------------------------------------------------------------------------- /tests/llamafy_qwen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/tests/llamafy_qwen.py -------------------------------------------------------------------------------- /tests/loftq_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/tests/loftq_init.py -------------------------------------------------------------------------------- /tests/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhiweihu1103/AgriMa/HEAD/tests/quantize.py --------------------------------------------------------------------------------