Fix convert_hf_checkpoint.py (#208)

As model has been moved to `build` folder by #192
pytorch · Jul 17, 2024 · d052e36 · d052e36
1 parent 1507940
commit d052e36
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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