point demo at ASSERT-KTH
Browse files- nla_demo.py +2 -2
nla_demo.py
CHANGED
|
@@ -3,7 +3,7 @@
|
|
| 3 |
|
| 4 |
pip install torch transformers accelerate huggingface_hub numpy pyyaml
|
| 5 |
hf auth login
|
| 6 |
-
hf download
|
| 7 |
|
| 8 |
python nla_demo.py --mode code 'def gcd(a, b):
|
| 9 |
while b: a, b = b, a % b
|
|
@@ -38,7 +38,7 @@ import numpy as np, torch, transformers, yaml
|
|
| 38 |
from huggingface_hub import hf_hub_download
|
| 39 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 40 |
|
| 41 |
-
REPO = "
|
| 42 |
CODE_SUB, TEXT_SUB = "code-decoder", "nla/actor"
|
| 43 |
BASE = "Qwen/Qwen3-8B"
|
| 44 |
# Verified per mode by A/B (job 17444219) -- do NOT "unify" these.
|
|
|
|
| 3 |
|
| 4 |
pip install torch transformers accelerate huggingface_hub numpy pyyaml
|
| 5 |
hf auth login
|
| 6 |
+
hf download ASSERT-KTH/tiny-nla nla_demo.py --local-dir .
|
| 7 |
|
| 8 |
python nla_demo.py --mode code 'def gcd(a, b):
|
| 9 |
while b: a, b = b, a % b
|
|
|
|
| 38 |
from huggingface_hub import hf_hub_download
|
| 39 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 40 |
|
| 41 |
+
REPO = "ASSERT-KTH/tiny-nla"
|
| 42 |
CODE_SUB, TEXT_SUB = "code-decoder", "nla/actor"
|
| 43 |
BASE = "Qwen/Qwen3-8B"
|
| 44 |
# Verified per mode by A/B (job 17444219) -- do NOT "unify" these.
|