Views
No views yet
attn_type="parallax"). This is the 0.6B-parameter model.| Parameters | ~0.6B |
| Layers | 28 |
| Hidden size | 1024 |
| Intermediate size | 3072 |
| Attention heads | 16 (8 KV heads, GQA) |
| Head dim | 128 |
| Vocab size | 151936 |
| Max sequence length | 4096 |
| Attention | parallax (attn_type="parallax") |
| Tokenizer | Qwen3 |
trust_remote_code=True
(requires a recent transformers with Qwen3 support):1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "YifeiZuo/Parallax-0.6B"
4tok = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
6
7prompt = "The quick brown fox"
8ids = tok(prompt, return_tensors="pt").input_ids
9out = model.generate(ids, max_new_tokens=32)
10print(tok.decode(out[0], skip_special_tokens=True))