diff options
| author | Void Agent <void@jayrup.hermes> | 2026-07-29 19:48:28 +0100 |
|---|---|---|
| committer | Void Agent <void@jayrup.hermes> | 2026-07-29 19:48:28 +0100 |
| commit | 437bd7d616885c89220f342a66e0004d524b2de8 (patch) | |
| tree | 10b9e403dc3071e4cfe77a8cdd19b1c458182415 | |
| parent | e4ce11b38f58a562c09cb9d28c37358058ff0225 (diff) | |
jlens_v2: fix import path for src/ -> root layout
| -rw-r--r-- | src/jlens_v2.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/jlens_v2.py b/src/jlens_v2.py index 1bacbc6..c9b7019 100644 --- a/src/jlens_v2.py +++ b/src/jlens_v2.py @@ -7,6 +7,9 @@ import torch.nn as nn import numpy as np import pickle import os +import os, sys +# jlens_v2.py is in src/, model.py is in repo root +sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from collections import defaultdict def load_model(checkpoint_path, device='cuda'): |