release v0.6.0 (real)

2024-03-25 23:37:48 +08:00 · 2024-03-25 23:37:48 +08:00 · ba70aca8fb
parent 98a42cbdaa
commit ba70aca8fb
1 changed files with 0 additions and 2 deletions
--- a/src/llmtuner/train/utils.py
+++ b/src/llmtuner/train/utils.py
@ -158,8 +158,6 @@ def _create_galore_optimizer(
    finetuning_args: "FinetuningArguments",
    max_steps: int,
 ) -> "torch.optim.Optimizer":
-    require_version("galore_torch", "To fix: pip install galore-torch")
-
    if len(finetuning_args.galore_target) == 1 and finetuning_args.galore_target[0] == "all":
        galore_targets = find_all_linear_modules(model)
    else: