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