diff --git a/src/llmtuner/train/utils.py b/src/llmtuner/train/utils.py index fd9c4795..0093f4a4 100644 --- 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: