From d052e36c49a0107a39d40e421f60bc5f37253c1b Mon Sep 17 00:00:00 2001
From: Nikita Shulga <2453524+malfet@users.noreply.github.com>
Date: Mon, 15 Apr 2024 16:46:02 -0700
Subject: [PATCH] Fix convert_hf_checkpoint.py (#208)

As model has been moved to `build` folder by https://github.com/pytorch/torchchat/pull/192
---
 scripts/convert_hf_checkpoint.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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