add adam_mini to readme

2024-08-09 20:02:03 +08:00
parent ef482394f0
commit e2a28f51c6
12 changed files with 94 additions and 34 deletions
--- a/src/llamafactory/hparams/parser.py
+++ b/src/llamafactory/hparams/parser.py
@@ -128,6 +128,9 @@ def _check_extra_dependencies(
    if finetuning_args.use_badam:
        require_version("badam>=1.2.1", "To fix: pip install badam>=1.2.1")

+    if finetuning_args.use_adam_mini:
+        require_version("adam-mini", "To fix: pip install adam-mini")
+
    if finetuning_args.plot_loss:
        require_version("matplotlib", "To fix: pip install matplotlib")