diff --git a/litgpt/scripts/merge_lora.py b/litgpt/scripts/merge_lora.py index e80a56c70a..aff59daef4 100644 --- a/litgpt/scripts/merge_lora.py +++ b/litgpt/scripts/merge_lora.py @@ -1,7 +1,6 @@ # Copyright Lightning AI. Licensed under the Apache License 2.0, see LICENSE file. """This script merges the LoRA weights with the base model""" -import re from pathlib import Path from typing import Any, Dict, Optional, Tuple