diff options
Diffstat (limited to 'src/jlens.py')
| -rw-r--r-- | src/jlens.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/jlens.py b/src/jlens.py index 16ff7ab..f332d44 100644 --- a/src/jlens.py +++ b/src/jlens.py @@ -346,9 +346,11 @@ if __name__ == '__main__': args = parser.parse_args() - # Import model from local dir + # Import model from project root import sys - sys.path.insert(0, os.path.dirname(__file__)) + # jlens.py is in src/, model.py is in repo root + project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) + sys.path.insert(0, project_root) from model import GPT, GPTConfig # Load model |