diff --git a/src/cerebras/modelzoo/tools/convert_checkpoint.py b/src/cerebras/modelzoo/tools/convert_checkpoint.py index f96fbb1..121cdf0 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,