point code demo at ASSERT-KTH
Browse files- 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
|
| 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="
|
| 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)
|