From 23ed090fb3e77d5283b52bc3738ff111fa4f4d54 Mon Sep 17 00:00:00 2001 From: jophinj Date: Fri, 14 Jun 2024 10:58:23 +0200 Subject: [PATCH] Fixes launch error of convert_checkpoint.py script Fixes the path issue that causes the error "ModuleNotFoundError: No module named 'cerebras.modelzoo'" while calling the convert_checkpoint.py script. --- src/cerebras/modelzoo/tools/convert_checkpoint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cerebras/modelzoo/tools/convert_checkpoint.py b/src/cerebras/modelzoo/tools/convert_checkpoint.py index f96fbb13..121cdf00 100644 --- a/src/cerebras/modelzoo/tools/convert_checkpoint.py +++ b/src/cerebras/modelzoo/tools/convert_checkpoint.py @@ -28,7 +28,7 @@ import cerebras.pytorch as cstorch -sys.path.append(os.path.join(os.path.dirname(__file__), "../")) +sys.path.append(os.path.join(os.path.dirname(__file__), "../../..")) from cerebras.modelzoo.tools.checkpoint_converters.base_converter import ( BaseCheckpointConverter, BaseConfigConverter,