diff --git a/src/sparseml/utils/helpers.py b/src/sparseml/utils/helpers.py index b7ffe9f0415..73f8f9d3cfa 100644 --- a/src/sparseml/utils/helpers.py +++ b/src/sparseml/utils/helpers.py @@ -72,7 +72,7 @@ FROM_PARAM_TOKEN = "__FROM_PARAM__" RECIPE_METADATA_KEY = "__metadata__" FRAMEWORK_METADATA_KEY = "framework_metadata" -ROOT_PATH = Path(__file__).resolve().parents[3] +ROOT_PATH = Path(__file__).resolve().parents[1] _LOGGER = logging.getLogger(__name__)