diff --git a/litgpt/__main__.py b/litgpt/__main__.py index 1c2db41eec..59d53ac904 100644 --- a/litgpt/__main__.py +++ b/litgpt/__main__.py @@ -23,7 +23,7 @@ ) from litgpt.scripts.download import download_from_hub as download_fn from litgpt.scripts.merge_lora import merge_lora as merge_lora_fn -from litgpt.scripts.evaluate import convert_and_evaluate as evaluate_fn +from litgpt.eval.evaluate import convert_and_evaluate as evaluate_fn if TYPE_CHECKING: from jsonargparse import ArgumentParser