Views
No views yet
| Base Model | SlitherCode/tiny-edu-166m |
| Architecture | ParchmentLM (LLaMA-style, tiktoken cl100k_base tokenizer) |
| Parameters | 166M |
| Pretraining Data | FineWeb (~4B tokens) |
| SFT Data | yahma/alpaca-cleaned (52k examples) |
| Training Epochs | 3 |
| Precision | bfloat16 |
1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4tokenizer = AutoTokenizer.from_pretrained("SlitherCode/tiny-edu-166m", trust_remote_code=True)
5model = AutoModelForCausalLM.from_pretrained("SlitherCode/tiny-edu-166m-instruct-v0", trust_remote_code=True)
6model.eval()
7
8messages = [
9 {"role": "system", "content": "You are a helpful assistant."},
10 {"role": "user", "content": "What is the capital of France?"}
11]
12
13prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
14inputs = tokenizer(prompt, return_tensors="pt")
15input_len = inputs["input_ids"].shape[1]
16
17with torch.no_grad():
18 outputs = model.generate(
19 **inputs,
20 max_new_tokens=200,
21 do_sample=False,
22 repetition_penalty=1.1,
23 eos_token_id=tokenizer.eos_token_id,
24 pad_token_id=tokenizer.eos_token_id,
25 )
26
27response = tokenizer.decode(outputs[0][input_len:], skip_special_tokens=True)
28print(response)<|endoftext|> as the turn separator:system
You are a helpful assistant.<|endoftext|>
user
What is the capital of France?<|endoftext|>
assistantTaori et al., "Alpaca: A Strong, Replicable Instruction-Following Model", Stanford University, 2023. Cleaned version: https://github.com/gururise/AlpacaDataCleaned