Tensor-1.5-small / README.md
VGribAI's picture
Update README.md
81ed35c verified
|
Raw History Blame Contribute Delete
2.51 kB
---
library_name: transformers
pipeline_tag: text-generation
license: apache-2.0
language:
- en
tags:
- text-generation
- conversational
- qwen2
- from-scratch
- english
---
# Tensor 1.5
Tensor 1.5 is a compact conversational language model built from scratch by **VGribAI**.
It opens the Tensor line: the first model whose name carries no parameter size, the first one
trained exclusively in English, and a complete assistant rather than a base checkpoint, since it
has gone through supervised fine-tuning.
![bencmarks](bench.jpg)
## What makes it different
- **No size in the name.** Tensor 1.5 starts a naming scheme where the version identifies the
model, not the parameter count.
- **English only.** The tokenizer, the pretraining and the fine-tuning are all English. The model
is not expected to follow other languages.
- **A finished assistant.** Tensor 1.5 follows the chat format, keeps the assistant role and stops
on its own turn end. It is a model you talk to, not a text continuation base.
## Architecture
| | |
|---|---|
| Decoder | Qwen2 style (12 layers, hidden 640, GQA 8 query / 2 KV heads, SwiGLU) |
| Parameters | 50.26M, tied embeddings |
| Vocabulary | 4000 byte-level BPE, trained for this model |
| Context | 512 tokens |
| Default answer length | 64 tokens |
| Weights | fp32 safetensors, plus a f16 GGUF build |
## Intended use
Short answers, quick summaries, simple rewrites and light chit-chat on weak hardware: CPUs,
old laptops, single-board computers and phones.
## Usage
```python
from transformers import AutoModelForCausalLM, AutoTokenizer
tok = AutoTokenizer.from_pretrained("tensor-1.5")
model = AutoModelForCausalLM.from_pretrained("tensor-1.5")
msgs = [{"role": "user", "content": "Who are you?"}]
ids = tok.apply_chat_template(msgs, add_generation_prompt=True, return_tensors="pt")
out = model.generate(ids, max_new_tokens=64, eos_token_id=[0, 2], pad_token_id=2)
print(tok.decode(out[0, ids.shape[1]:], skip_special_tokens=True))
```
## Chat format
```
<|im_start|>system
You are Tensor, a small assistant made by VGribAI.<|im_end|>
<|im_start|>user
Hello!<|im_end|>
<|im_start|>assistant
Hello! How can I help you?<|im_end|>
```
Stop tokens: `<|im_end|>` (id 2) and `<|eos|>` (id 0).
## Limits
- English only.
- 512 tokens of context, 64 tokens per answer.
- A very small model: fluent phrasing, but no reliable factual knowledge, arithmetic or long
reasoning. Treat answers as text, not as facts.
- No memory beyond the current context window.