Views
No views yet
| Base model | SmolLM2-135M |
| Dataset | smoltalk/everyday-conversations (2k samples) |
| Epochs | 1 |
| LoRA r | 8 |
| LoRA alpha | 16 |
| Target modules | q_proj, v_proj |
| max_seq_length | 1024 |
| Compute | Google Colab T4 (~15 min) |
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3import torch
4
5BASE_ID = "HuggingFaceTB/SmolLM2-135M"
6ADAPTER_ID = "hfm8tr/smollm2-135m-smoltalk-lora"
7
8tokenizer = AutoTokenizer.from_pretrained(ADAPTER_ID)
9base = AutoModelForCausalLM.from_pretrained(BASE_ID, torch_dtype=torch.float32)
10model = PeftModel.from_pretrained(base, ADAPTER_ID)
11model.eval()
12
13messages = [{"role": "user", "content": "What is gravity?"}]
14prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
15inputs = tokenizer(prompt, return_tensors="pt")
16
17with torch.no_grad():
18 out = model.generate(**inputs, max_new_tokens=100, do_sample=True, temperature=0.3,
19 pad_token_id=tokenizer.eos_token_id)
20print(tokenizer.decode(out[0][inputs['input_ids'].shape[1]:], skip_special_tokens=True))