release v0.6.0 (real)

hiyouga · Mar 25, 2024 · ba70aca · ba70aca
1 parent 98a42cb
commit ba70aca
Showing 1 changed file with 0 additions and 2 deletions.
diff --git 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: