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

point demo at ASSERT-KTH

Browse files
Files changed (1) hide show
  1. 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 TuHan/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,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 = "TuHan/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.
 
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.