diff --git a/scripts/convert_hf_checkpoint.py b/scripts/convert_hf_checkpoint.py index 5cd3d3ca3..428c4a733 100644 --- a/scripts/convert_hf_checkpoint.py +++ b/scripts/convert_hf_checkpoint.py @@ -12,8 +12,9 @@ import torch # support running without installing as a package -wd = Path(__file__).parent.parent.resolve() -sys.path.append(str(wd)) +wd = Path(__file__).parent.parent +sys.path.append(str(wd.resolve())) +sys.path.append(str((wd / "build").resolve())) from model import ModelArgs