Dmitriy-Zemskov commited on
Commit
36fe8b9
·
verified ·
1 Parent(s): 27163d5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -22
README.md CHANGED
@@ -18,25 +18,4 @@ The first generation of the CalmaCatMath series. This is an ultra-compact langua
18
  ### Model Features
19
  - **Size:** ~19M parameters — runs lightning-fast even on a potato.
20
  - **Vocabulary:** Custom Char tokenzier.
21
- - **Lore:** Basic NLP, Basic Arithmetic.
22
-
23
- ## How to run
24
-
25
- ```python
26
- from transformers import GPT2LMHeadModel
27
- import torch
28
-
29
- from tokenizer import CharTokenizer
30
-
31
- model = GPT2LMHeadModel.from_pretrained("ViorikaAI-org/CalmaCatMath")
32
- tokenizer = CharTokenizer()
33
-
34
- text = "<s> 2+2="
35
-
36
- ids = torch.tensor([tokenizer.encode(text)])
37
-
38
- with torch.no_grad():
39
- out = model.generate(ids, max_new_tokens=16)
40
-
41
- print(tokenizer.decode(out[0].tolist()))
42
- ```
 
18
  ### Model Features
19
  - **Size:** ~19M parameters — runs lightning-fast even on a potato.
20
  - **Vocabulary:** Custom Char tokenzier.
21
+ - **Lore:** Basic NLP, Basic Arithmetic.