-
- Downloads
Merge pull request #3287 from Ledzy/badam
[Feature] Add BAdam algorithm
Showing
- examples/extras/badam/sft.sh 35 additions, 0 deletionsexamples/extras/badam/sft.sh
- setup.py 1 addition, 0 deletionssetup.py
- src/llmtuner/hparams/finetuning_args.py 49 additions, 2 deletionssrc/llmtuner/hparams/finetuning_args.py
- src/llmtuner/hparams/parser.py 11 additions, 1 deletionsrc/llmtuner/hparams/parser.py
- src/llmtuner/model/adapter.py 3 additions, 3 deletionssrc/llmtuner/model/adapter.py
- src/llmtuner/model/patcher.py 2 additions, 2 deletionssrc/llmtuner/model/patcher.py
- src/llmtuner/model/utils.py 33 additions, 1 deletionsrc/llmtuner/model/utils.py
- src/llmtuner/train/sft/trainer.py 5 additions, 0 deletionssrc/llmtuner/train/sft/trainer.py
- src/llmtuner/train/utils.py 76 additions, 2 deletionssrc/llmtuner/train/utils.py
Loading
Please register or sign in to comment