TuHan commited on
Commit
1eba213
·
verified ·
1 Parent(s): 3291d81

point code demo at ASSERT-KTH

Browse files
Files changed (1) hide show
  1. nla_code_demo.py +2 -2
nla_code_demo.py CHANGED
@@ -3,7 +3,7 @@
3
 
4
  pip install torch transformers huggingface_hub numpy pyyaml
5
  hf auth login
6
- hf download TuHan/tiny-nla nla_code_demo.py --local-dir .
7
  python nla_code_demo.py --code 'def is_palindrome(s):
8
  s = s.lower()
9
  return s == s[::-1]'
@@ -91,7 +91,7 @@ def main():
91
  g = p.add_mutually_exclusive_group(required=True)
92
  g.add_argument("--code", help="function source")
93
  g.add_argument("--file", help="path to a .py file")
94
- p.add_argument("--repo", default="TuHan/tiny-nla")
95
  p.add_argument("--subfolder", default="code-decoder")
96
  p.add_argument("--base-model", default="Qwen/Qwen3-8B")
97
  p.add_argument("--injection-scale", type=float, default=300.0)
 
3
 
4
  pip install torch transformers huggingface_hub numpy pyyaml
5
  hf auth login
6
+ hf download ASSERT-KTH/tiny-nla nla_code_demo.py --local-dir .
7
  python nla_code_demo.py --code 'def is_palindrome(s):
8
  s = s.lower()
9
  return s == s[::-1]'
 
91
  g = p.add_mutually_exclusive_group(required=True)
92
  g.add_argument("--code", help="function source")
93
  g.add_argument("--file", help="path to a .py file")
94
+ p.add_argument("--repo", default="ASSERT-KTH/tiny-nla")
95
  p.add_argument("--subfolder", default="code-decoder")
96
  p.add_argument("--base-model", default="Qwen/Qwen3-8B")
97
  p.add_argument("--injection-scale", type=float, default=300.0)