diff options
Diffstat (limited to 'src/synthetic_pair.py')
| -rw-r--r-- | src/synthetic_pair.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/synthetic_pair.py b/src/synthetic_pair.py index 7614fbe..6c6c722 100644 --- a/src/synthetic_pair.py +++ b/src/synthetic_pair.py @@ -32,6 +32,7 @@ DATA_SRC = 'data/shakespeare_char/input.txt' DATA_DIR = 'data/synth_pair' OUT_ROOT = 'out-synth-pair' JLENS_OUT = 'outputs/synth_pair' +REPO_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) T_STRUCT = '@' T_NOISE = '#' TRIGGER = 'the ' @@ -164,7 +165,7 @@ def jlens(seed, layers, n_prompts): "--chunk", "16", "--output_dir", out_dir] print("running:", " ".join(cmd)) - r = subprocess.run(cmd, cwd='/workspace/code') + r = subprocess.run(cmd, cwd=REPO_ROOT) assert r.returncode == 0, "jlens_v3 failed" |